华为OJ——称砝码

称砝码

题目描述

现有一组砝码，重量互不相等，分别为m1,m2,m3…mn；

每种砝码对应的数量为x1,x2,x3...xn。现在要用这些砝码去称物体的重量，问能称出多少中不同的重量。

注：

称重重量包括0

方法原型：public static int fama(int n, int[] weight, int[] nums)

[b]输入描述:[/b]
输入包含多组测试数据。
对于每组测试数据：
第一行：n ---砝码数(范围[1,10])
第二行：m1 m2 m3 ... mn ---每个砝码的重量(范围[1,2000])
第三行：x1 x2 x3 .... xn ---每个砝码的数量(范围[1,6])
[b]输出描述:[/b]
利用给定的砝码可以称出的不同的重量数

[b]输入例子:[/b]
2
1 2
2 1

[b]输出例子:[/b]
5
解答代码：

#include<iostream>
#include<vector>
#include<cstdio>
using namespace std;

int main()
{
int i,j,k,n;
while(cin>>n)
{
vector<int> dataWeight(n+1);
vector<int> dataNum(n+1);
//输入各砝码质量
for(i=1; i<=n; i++)
cin>>dataWeight[i];
//输入个砝码质量
for(i=1; i<=n; i++)
cin>>dataNum[i];

int maxWeight=0;
for(i=1; i<=n; i++)
maxWeight+=dataWeight[i]*dataNum[i];

//构造judge表,表的大小为maxWeight
vector<int> judge(maxWeight+1,0);
judge[0]=1;//可以称量质量为0的物体
//构造表
for(i=1; i<=n; i++)
for(j=maxWeight; j>0; j--)
for(k=1; k<=dataNum[i]; k++)
if(j-k*dataWeight[i] >= 0)
judge[j] = judge[j] | judge[j-k*dataWeight[i]];
int count=0;
for(i=0; i<maxWeight+1; i++)
{
if(judge[i])
count++;
}
cout<<count<<endl;
}
return 0;
}