Description
夏令营有N个人,每个人的力气为M(i)。请大家从这N个人中选出若干人,如果这些人可以分成两组且两组力气之和完全相等,则称为一个合法的选法,问有多少种合法的选法?
数据范围2<=N<=20,1<=M(i)<=100000000
Analysis
仅观察数据范围,得出结论:枚举和是不可能的!
最简单的想法
3N
枚举。
仍旧考虑暴力,可以应用Meet-in-the-middle即折半搜索
分别考虑两边的结果,合并统计答案
设左组选了
a,b...
,右组选了
c,d...
则有
a+b+...=c+d+...
就有
a−c+...=−b+d+...
然后就切了。
时间复杂度约为
O(Ans)
记得判重,注意卡常!
Code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define rep(i,x) for(int i=ls[x];i;i=nx[i])
#define MAX 22
#define M 1048577
#define S 1025
#define NN 44
using namespace std;
int Z,N,n;
int a[MAX],h[M],er[MAX];
int ans=0;
bool bz[S][S];
int nx[M],ls[M],to[M],num=0;
int hash(int x)
{
int p=(x%M+M)%M;
while(h[p]!=Z && h[p]!=x) p=(p+1)%M;
return p;
}
void link(int x,int y)
{
num++;
to[num]=y;
nx[num]=ls[x];
ls[x]=num;
}
void dfs(int x,int s,int st)
{
if(x>n)
{
int t=hash(s);
h[t]=s;
bool tf=true;
rep(i,t)
{
int v=to[i];
if(v==st)
{
tf=false;
break;
}
}
if(tf) link(t,st);
return;
}
fo(i,-1,1)
{
int p=st;
if(i) p+=er[x-1];
dfs(x+1,s+a[x]*i,p);
}
}
void get(int x,int s,int st)
{
if(x>N)
{
int t=hash(s);
if(h[t]==Z) return;
rep(i,t)
{
int v=to[i];
if(!bz[st][v])
{
bz[st][v]=true;
ans++;
}
}
return;
}
fo(i,-1,1)
{
int p=st;
if(i) p+=er[x-n-1];
get(x+1,s+a[x]*i,p);
}
}
int main()
{
freopen("subset.in","r",stdin);
freopen("subset.out","w",stdout);
scanf("%d",&N);
n=N/2;
fo(i,1,N) scanf("%d",&a[i]);
memset(h,128,sizeof(h));
Z=h[1];
er[0]=1;
fo(i,1,MAX-1) er[i]=er[i-1]*2;
dfs(1,0,0);
get(n+1,0,0);
printf("%d",ans-1);
}