codeforces/gym/101291/B
题意:给你n个杠铃的杆子,在给你m个杠铃片,问你能组成多少个重量不同的完整杠铃(杠铃杆子也算一个完整的的杠铃)
解题思路:dfs直接搜,数据很小,每个杠铃片有三种状态(放杆子左边,放杆子右边,两边都不放),用set存一下,去下重,正好要从小到大序输出
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | #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; } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步