引入
今天听了大佬的讲课,然后打了一下砝码称重这道题。
我用了两种方法:枚举法和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++问题时,要注意到前后匹配这一点。