首页 > 代码库 > 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: );    }}
View Code

 

CC ANUMLA(STL的运用)