传送门:http://www.oj.swust.edu.cn/problem/show/2469
应该还有更好的剪枝,不过还没想到。。。
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n,a[20];
char x[20];
bool flag;
int ans;
void dfs(int k)
{
if(k>n-1) return ;
if(k==n-1&&!flag)
{
int i=0,sum1=a[0],sum2=0;
for(;x[i]!='='&&i<n-1;i++)
{
if(x[i]=='+') sum1+=a[i+1];
else sum1-=a[i+1];
}
for(;i<n-1;i++)
{
if(x[i]=='+') sum2+=a[i+1];
else if(x[i]=='-') sum2-=a[i+1];
else sum2=a[i+1];
}
if(sum2==sum1)
ans++;
return ;
}
x[k]='+';
dfs(k+1);
x[k]='-';
dfs(k+1);
if(flag)
{
flag=false;
x[k]='=';
dfs(k+1);
flag=true;
}
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
flag=true; ans=0;
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
dfs(0);
printf("%d\n",ans);
}
return 0;
}