swust oj 2469: 小Y的难题1

传送门：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;
}