华为OJ——称砝码

称砝码

题目描述

现有一组砝码，重量互不相等，分别为m1,m2,m3…mn；
每种砝码对应的数量为x1,x2,x3...xn。现在要用这些砝码去称物体的重量，问能称出多少中不同的重量。

注：

称重重量包括0

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

输入描述:

输入包含多组测试数据。

对于每组测试数据：

第一行：n ---砝码数(范围[1,10])

第二行：m1 m2 m3 ... mn ---每个砝码的重量(范围[1,2000])

第三行：x1 x2 x3 .... xn ---每个砝码的数量(范围[1,6])

输出描述:

利用给定的砝码可以称出的不同的重量数

输入例子:

2

1 2

2 1

 

输出例子:

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;
}