P2347 砝码称重

P2347 砝码称重

题目链接

题目描述

设有1g、2g、3g、5g、10g、20g的砝码各若干枚（其总重≤1000），

输入格式

输入方式：a1 a2 a3 a4 a5 a6
（表示1g砝码有a1个，2g砝码有a2个，…，20g砝码有a6个）

输出格式

输出方式：Total=N
（N表示用这些砝码能称出的不同重量的个数，但不包括一个砝码也不用的情况）

输入样例

1 1 0 0 0 0

输出样例

Total=3

题目分析

多重背包的问题,1000以内的数据可以用二进制优化,优化成01背包的问题,1000以上的要用单调队列优化了,这题的数据比较小就用二进制优化了.二进制优化就是将个数分成很多块,用的时候取其中的块就行了
7可以分成1 2 4 .
12可以分为1 2 4 5
这样里面的数都可以用二进制的加法来表示每一个数了

代码实现

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<map>
#include<set> 
#include<string>
#include<stack>
#include<queue>
#include<vector>
#include<math.h> 
using namespace std;
typedef long long int ll;
const int maxn=2e5+5;
const int inf=1e8;
ll dp[maxn];//dp[i]表示所给的砝码可以表示i质量
int vv[maxn];
int ww[maxn]; 
int n;
int main()
{
	int num=1;
	int s;
	int sum=0;
	int a[7]={0,1,2,3,5,10,20};//质量分组
	memset(dp,0,sizeof(dp));
	for(int i=1;i<=6;i++)//6个就遍历6次
	{
		cin>>s;//输入数量
		sum+=s*a[i];//sum用来计算质量的最大值
		int v=a[i];
		//下面是二进制优化
		for(int j=1;j<=s;j<<=1)//j<<=1,是j*2的意思,比j*=的速度更快
		{
			vv[num++]=j*v;//记录二进制后新的价值
			s-=j;//加后减去
		}
		if(s)//如果还有剩余不能用二进制来存放,就单独放入最后一个数组
		{
			vv[num++]=s*v;
		} 
	}
	dp[0]=1;//dp[0]一定要等于1,当背包可以容纳下,dp[j-vv[i]]中会减成dp[0],dp[0]=1才能让数量增加.
	for(int i=1;i<num;i++)//num为二进制后的新的质量
	{
		for(int j=sum;j>=vv[i];j--)
		{
			dp[j]+=dp[j-vv[i]];//数量相加
		}
	}
	int cnt=0;
	for(int i=1;i<=sum;i++)//dp[i]表示所给的砝码可以表示i质量,所以如果可以到达数量就加一
		if(dp[i])
			cnt++;
	cout<<"Total="<<cnt<<endl;
}