codeforces/gym/101291/B

题意:给你n个杠铃的杆子,在给你m个杠铃片,问你能组成多少个重量不同的完整杠铃(杠铃杆子也算一个完整的的杠铃)

解题思路:dfs直接搜,数据很小,每个杠铃片有三种状态(放杆子左边,放杆子右边,两边都不放),用set存一下,去下重,正好要从小到大序输出

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<set>
#define ll long long
using namespace std;
int a[20];
int b[20];
int n,m;
int cnt=0;
set<ll>q;
void dfs(ll x,ll l,ll r)
{
    if(x>m)
        return;
    if(l==r)
    {
        q.insert(r+l);
    }
    dfs(x+1,l+a[x+1],r);
    dfs(x+1,l,r+a[x+1]);
    dfs(x+1,l,r);
}
int main()
{
    set<ll>ans;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>b[i];
    for(int i=1;i<=m;i++)
        cin>>a[i];
    a[0]=0;
    dfs(0,0,0);
    for(int i=1;i<=n;i++)
    {
        for(set<ll>::iterator it=q.begin();it!=q.end();it++)
        {
            ans.insert(*it+b[i]);
        }
    }
    for(set<ll>::iterator it=ans.begin();it!=ans.end();it++)
    {
        cout<<*it<<endl;
    }
    return 0;
}

  

posted @ 2018-04-10 19:37  荒岛的龟  阅读(228)  评论(0编辑  收藏  举报