Barbells（三进制枚举）-CSDN博客

本文链接：https://blog.csdn.net/gml1999/article/details/89530986

题意：就是给你几个哑铃片几个空杆让你组合不同的重量

题解：枚举；得想到三进制枚举；每片哑铃都有三种状态：加在左边，加在右边，内边都不加

(也犯了谋篇题解的错误 long long 取模要比 int 取模慢会超时)

#include<cstdio>
#include<cstring>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<stdlib.h>
#include<set>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=1000000;
typedef long long ll;
ll a[20];
ll b[20];
set<ll> s;
int  main(){
	 int n;
	 int m;
	 scanf("%d%d",&n,&m);
	 ll tmp=1;//三进制枚举 三个状态 
	 for(int i=0;i<n;i++)
	   scanf("%lld",&a[i]);
	 for(int i=0;i<m;i++){
	 	scanf("%lld",&b[i]);
	 	tmp=tmp*3;//对于每一个哑铃片都有三个状态 
	 }
	 for(int i=0;i<tmp;i++){
	 	ll na=0;
	 	ll nb=0;
	 	int k=0;
	 	int t=i;//long long t 就会T    
	 	while(t){
	 		if(t%3==1)  na+=b[k];//该位为1则加在左边 
	 		if(t%3==2)  nb+=b[k];//改为为2则加在右边 
	 		k++;
	 		t=t/3;
		 }
		 if(na==nb){
		 	for(int j=0;j<n;j++){
		 		ll ans=na+nb+a[j];
		 		s.insert(ans);
			 }	
		 } 
	 }
	  set<ll>:: iterator it;
	  for(it=s.begin();it!=s.end();it++){
	  	  printf("%lld\n",*it);
	  } 
}