首页 > 代码库 > CC ANUMLA(STL的运用)
CC ANUMLA(STL的运用)
题目连接:http://www.codechef.com/problems/ANUMLA
题意:给一个序列所有子集和(2^n个子集),复原这个序列。。。
如:0 1 1 2 2 3 3 4 原序列为1 1 2
分析:每次找出最小的那个元素,再删除掉可能由该元素相加得到的元素,如上面那个例子,将所有可能相加得到其他元素存在一个数组add中,0是空集,先去掉,剩下第一个元素1必定是原序列最小元素,取出答案数组ans中,一步步下去。
1.取出1放进ans数组中,ans数组:1 相加辅助数组add:1,删除1后原序列:1 2 2 3 3 4
2.取出最小元素1放进ans数组,ans数组1 1 相加辅助数组add:1 1 2(1+1,1),删除1,2后序列:2 3 3 4
3.取出最小元素2放进ans数组,ans数组1 1 2 相加辅助数组add:1 1 2 2 3 3 4(1+2,1+2,2+2,2),删除2,3,3,4后原序列已空。
同样序列0 2 5 7 10 12 15 17去掉0,原序列为 2 5 7 10 12 15 17
1.取出1放进ans数组中,ans数组:2 相加辅助数组add:2,删除2后原序列: 5 7 10 12 15 17
2.取出最小元素5放进ans数组,ans数组2 5 相加辅助数组add:2,5,7(2+5,5),删除5,7后序列:10 12 15 17
3.取出最小元素10放进ans数组,ans数组2 5 10 相加辅助数组add:2 5 7 10 12 15 17(10+2,10+5,10+7,10),删除2,3,3,4后原序列已空。
这题用STL中的multiset处理起来太方便了。
#include<bits/stdc++.h>using namespace std;multiset<int>Set;vector<int>ans,add;int main(){ int T,n,x; scanf("%d",&T); while(T--) { scanf("%d",&n); Set.clear();ans.clear();add.clear(); for(int i=0;i<(1<<n);i++) { scanf("%d",&x); Set.insert(x); } Set.erase(0); x=n; while(x--) { int mn=*(Set.begin());//取出最小元素放进ans数组 ans.push_back(mn); for(int i=0,sz=add.size();i<sz;i++) { int ad=add[i]+mn;add.push_back(ad); Set.erase(Set.find(ad));//删除由各个元素相加得到的子集 } add.push_back(mn); Set.erase(Set.find(mn)); } for(int i=0,sz=ans.size();i<sz;i++)printf("%d%c",ans[i],(i==n-1)?‘\n‘:‘ ‘); }}
CC ANUMLA(STL的运用)