【华为OJ】称砝码

描述	现有一组砝码，重量互不相等，分别为m1、m2……mn；他们可取的最大数量分别为x1、x2……xn。现在要用这些砝码去称物体的重量，问能称出多少中不同的重量。   注： 称重重量包括0 要对输入数据进行校验   方法原型：public static int fama(int n, int[] weight, int[] nums)
知识点	字符串,循环,函数,指针,枚举,位运算,结构体,联合体,文件操作,递归
运行时间限制	10M
内存限制	128
输入	int n：n表示有多少组重量不同的砝码，1<=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<> int[] weight：表示n组砝码的重量，1<=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<> int[] num：表示n组砝码的最大数量，1<=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<>
输出	利用给定的砝码可以称出的不同的重量数
样例输入	2 1 2 2 1
样例输出	5

思路

源代码

1. #include <iostream>            
2. #include <vector>            
3. using namespace std;  
4. int fama(int n, int weight[], int nums[]){  
5.     //定义bool型标记数组，记录该重量是否存在            
6.     //计算标记数组的最大可能的不同重量数，分配空间            
7.     int maxWeight = 0;  
8.     for (int i = 0; i < n; i++)  
9.         maxWeight += weight[i] * nums[i];  
10.     vector<bool> flag(maxWeight + 1, false);  
11.     //标记数组的初始状态            
12.     flag[0] = true;//没有砝码的情况，称重重量为0，也属于一种情况            
13.     int total = 0;  
14.     int count = 1;//没有砝码的情况，算一种            
15.   
16.     //从第1组砝码开始添加            
17.     int sum = 0;  
18.     for (int i = 0; i < n; i++){//循环添加第i组砝码            
19.         for (int j = 1; j <= nums[i]; j++){//第i组砝码，逐个添加            
20.             for (int k = 0; k <= total; k++){  
21.                 if (flag[k]){  
22.                     sum = k + weight[i] * j;  
23.                     if (!flag[sum]){  
24.                         flag[sum] = true;  
25.                         count++;  
26.                     }  
27.                 }  
28.             }  
29.         }  
30.         total = sum;  
31.     }  
32.     return count;  
33. }  
34. void main(){  
35.     int n;  
36.     cin >> n;  
37.     int* weight = new int[n];  
38.     int* nums = new int[n];  
39.     for (int i = 0; i < n; i++)  
40.         cin >> weight[i];  
41.     for (int i = 0; i < n; i++)  
42.         cin >> nums[i];  
43.     cout << fama(n, weight, nums) << endl;  
44.       
45.     delete[] weight;  
46.     delete[] nums;  
47. }