题目描述
现有一组砝码,重量互不相等,分别为m1,m2,m3…mn;
每种砝码对应的数量为x1,x2,x3…xn。现在要用这些砝码去称物体的重量,问能称出多少中不同的重量。
注:
称重重量包括0
输入描述:
输入包含多组测试数据。
对于每组测试数据:
第一行:n — 砝码数(范围[1,10])
第二行:m1 m2 m3 … mn — 每个砝码的重量(范围[1,2000])
第三行:x1 x2 x3 …. xn — 每个砝码的数量(范围[1,6])
输出描述:
利用给定的砝码可以称出的不同的重量数
输入例子:
2
1 2
2 1
输出例子:
5
算法思想
从砝码1开始分析,假设前i个砝码能称出的不同重量为Q[i],那么Q[i]一定是这样计算出来的:在Q[i-1]的基础上,对Q[i-1]个不同的重量,分别添加k个砝码i,再添加的过程中除去重复情况。
假设:w[N]表示N个不同重量的砝码(例子中N=6),w[0~N-1]。
c[N]表示N个不同砝码相应的数量,c[1~N]。
则:Q[i] = (Q[i-1] + k*w[i])-添加过程中重复的个数。其中0=
代码块
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
void func19(int n, int m[], int num[])
{
vector<int> weight;
weight.push_back(0);
for (int i = 1; i <= num[0]; ++i)//第一个砝码
{
weight.push_back(i * m[0]);
}
for (int i = 1; i < n; ++i) //前n个砝码
{
int len = weight.size();
for (int j = 1; j <= num[i]; ++j)//砝码个数递增
{
for (int k = 0; k < len; ++k)//之前所有可测重量
{
int wei = weight[k] + j*m[i];
if (find(weight.begin(), weight.end(), wei) == weight.end())
{
weight.push_back(wei);
}
}
}
}
cout << weight.size() << endl;
}
int main()
{
int n;
int m[10], num[10];
while (cin >> n)
{
for (int i = 0; i < n; ++i)
{
cin >> m[i];
}
for (int i = 0; i < n; ++i)
{
cin >> num[i];
}
func19(n, m, num);
}
return 0;
}