华为机试--称砝码

现有一组砝码，重量互不相等，分别为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

//巧妙利用STL-set的无重复元素特性,来进行简单的模拟
#include <iostream>
#include <set>
using namespace std;

int n;
int m[2100], x[10];

int main()
{
    while(scanf("%d",&n) != EOF)
    {
        int sum = 0;

        for(int i=0;i<n;i++)
            scanf("%d",&m[i]);
        for(int i=0;i<n;i++)
            scanf("%d",&x[i]);
        for(int i=0;i<n;i++)
            sum += m[i]*x[i];//设置和的上限

        set<int> s;
        set<int>::iterator it;
        s.insert(sum);

        for(int i=0;i<n;i++)
        {
            it = s.begin();
            for(;it!=s.end();it++)
                for(int j=0;j<=x[i];j++)
                    if(*it >= j*m[i])
                        s.insert(*it-j*m[i]);
        }
        cout<<s.size()<<endl;
    }
    return 0;
}