洛谷——砝码称重

题目传送门：

引入

今天听了大佬的讲课，然后打了一下砝码称重这道题。
我用了两种方法：枚举法和01背包
枚举法在洛谷上只得了64分有4个超时了，01背包就全部通过了。
题目
题目描述
设有1g、2g、3g、5g、10g、20g的砝码各若干枚（其总重 \le 1000≤1000），

输入格式：输入六个数：a1 a2 a3 a4 a5 a6
输出方式：Total=N

• 输入输出样例
• 输入 1 1 0 0 0 0
• 输出 Total=3
  法一：枚举法就是相当于桶排序，就是用一个sum存储给定砝码能组成的所有的重量和，把他们用一个bool数组分别标记，最后遍历出所有的重量，统计所有给定砝码可以表示出来的重量。

//int flag[1005] ;
//int main() {
//	int a1, a2, a3, a4, a5, a6;
//	cin >> a1 >> a2 >> a3 >> a4 >> a5 >> a6;
//	//暴力枚举
//	
//	for (int i = 0; i <= a1; i++) {
//		for (int j = 0; j <= a2; j++) {
//			for (int k = 0; k <= a3; k++) {
//				for (int m = 0; m <= a4; m++) {
//					for (int l = 0; l <= a5; l++) {
//						for (int n = 0; n <= a6; n++) {
//							int sum = 0;
//							sum = 1 * i + 2 * j + 3 * k + 5 * m + 10 * l + 20 * n;
//							flag[sum] = true;
//						}
//					}
//				}
//			}
//		}
//	}
//	int count = 0;
//	for (int i = 1; i <= 1000; i++) {
//		if (flag[i]) count++;
//	}cout << "Total=" << count << endl;
//}

法二：01背包

• 先是准备好一个砝码序列，用来存放各种砝码 比如我输入的是2,1,2,1,1,1 ，最终我要用一个数组将它转换成1,1,2,3,3,5,10,20，这就是我要的砝码数列。
• 我要用f[i][j]表示我用了i个砝码，他们的总质量为j.
  动态规划的初始化f[k][0] = 1,也就是我一个砝码也没用，这种情况是存在的，故将其定义为1；
  状态转移方程：
• 如果我用了i -1 个砝码就得到了我想要的重量j ，那么我就将f[i][j]（我用i个砝码，第i个砝码我就去砝码数列中的0就ok了）定为1，表示我用i个所给的砝码可以把你给我的重量表示出来。也就是f[i][j] = 1;
• 如果我用了i个砝码才达到我想要的重量j，那么此时我用第i - 1个砝码可以达到重量j - w[i]，表示为f[i-1][j - w[i]]， 这样的话，说明如果f[i - 1][j - w[i]] = 1，那么我同样可以用i 个所给的砝码把你给我的重量表示出来。也就是f[i][j] = 1;
  综合以上的两种情况，有以下表达式成立：
  f[i][j] = (f[i - 1][j]) || (f[i - 1][j - w[i]])
  这是很显然的，当然为了让上式合法，我要分类讨论（j >= w[i]时上式完全合法，当j < w[i]时，上式改为 f[i][j] = (f[i - 1][j])
  最后，统计用这些砝码能称出的不同重量的个数即可。

int f[1005][1005] ;
int main() {
	int num = 0; //砝码的总数量
	int w[1005]; //砝码序列
	int a[7] = { 0,1, 2, 3, 5 ,10, 20 };//砝码对应的重量
	for (int i = 1; i < 7; i++) {
		int x;
		cin >> x;
		//构建序列2,1,2,1,1,1 ——> 1,1,2,3,3,5,10,20
		for(int j = 1; j <= x; j++)
		w[++num] = a[i];
	}
	//初始化动态规划，动态规划填表
	for (int k = 0; k <= 1000; k++) f[k][0] = 1;
	for (int i = 1; i <= num ; i++) {
		for (int j = 1; j <= 1000; j++) {
			if (j >= w[i])
				f[i][j] = (f[i - 1][j]) || (f[i - 1][j - w[i]]);
			else f[i][j] = f[i - 1][j];
		}
	}
	int cnt = 0;
	for (int i = 1; i <= 1000; i++) {
		if (f[num][i]) cnt++;
	}
	cout << "Total=" << cnt << endl;
}

这个代码可以通过洛谷上所有的案例。
最后还想说一个问题，就是上面代码中的一个注意点

for(int j = 1; j <= x; j++)
		w[++num] = a[i];

其中，w[++num]不能改为w[num++]
++ i 是先加后赋值；i ++ 是先赋值后加。
一般来说，这一点的不同不会造成太大的影响，但是这题必须注意，因为此处的num从0开始，w[++num]就相当于先给w[1]赋值为a[1]，刚好对应了下文中的状态转移部分。而如果是w[num++]， 它是先给w[0]赋值为a[1]，最终的w[num-1] = a[6]，到了状态转移时,会发现w[0] 匹配了a[1],w[num- 1]匹配了a[5],最终的a[6]却匹配w[num] (0）了，因此，这对结果造成了很大的影响。所以，以后在遇到++i 和 i++问题时，要注意到前后匹配这一点。