怎么去重,用set集合。
- 首先根据输入顺序,将砝码用数字序列表示,例如2个1g和1个2g,就用 1 1 2的序列表示;
- set序列用来表示加入当前砝码之前能产生的重量种类;
- set初始化为{0};当第一个1g砝码放入时,则set中需要插入原先set中的所有元素+1g后的结果;即{0, 0+1};
- 当第二个1g加入时,则set会插入{0+1, 1+1},就变成了{0, 1, 2};
- 重复上述步骤加入所有砝码;则最后set的大小即为能产生的重量种类。
#include <iostream>
#include <vector>
#include <set>
using namespace std;
int main()
{
int n, a[10], tmp;
while (cin >> n)
{
vector<int> v;
set<int> s;
for(int i = 0; i < n; i++) cin >> a[i];
for(int i = 0; i < n; i++)
{
cin >> tmp;
for(int j = 0; j < tmp; j++) v.push_back(a[i]);
}
s.insert(0);
for(int i = 0; i < v.size(); i++)
{
set<int> t(s);
for(auto it = t.begin(); it != t.end(); it ++)
{
s.insert(*it + v[i]);
}
}
cout << s.size() << endl;
}
return 0;
} 
京公网安备 11010502036488号