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

 

posted on 2015-01-28 01:04  lienus  阅读(165)  评论(0编辑  收藏  举报

导航