SSL2669 2017年8月7日提高组T1 选数(dfs)

2017年8月7日提高组T1 选数

Description

给出n个数a[i]，现在可以在其中任意选出若干个数，问有多少种选择方案，使得这几个数可以分成两个和相等的集合。

Input

第一行是一个正整数n，第二行每行n个正整数。

Output

输出一个数，表示方案数。

分析：每个数的系数有三种情况，分别是0,1,-1，0代表不选，1代表放入集合A，-1代表放入集合B。那么我们可以把序列分成长度相等的两部分，分别O(3^(n/2))搜索，把所有组合后的结果记录到两个数组，排序后用双指针统计答案即可。

代码

#include <cstdio>
#include <algorithm>
#include <math.h>
#define maxn 100000
using namespace std;

struct arr1
{
    int o,q;
}a[maxn];
struct arr2
{
    int o,q;
}b[maxn];

int c[maxn];
int n,p1,p2,p3;
bool fl,v[2000000];
long long ans;

void dfs1(int dep,int s,int u)
{
    if (s!=0) fl=true;
    if (fl&&dep>n/2)
    {
        p1++;
        a[p1].o=s;
        a[p1].q=u;
        if (s==0&&!v[u])
        {
            ans++;
            v[u]=true;
        }
        return;
    }
    if (dep>n/2) return;
    dfs1(dep+1,s,u);
    dfs1(dep+1,s+c[dep],u+pow(2,dep-1));
    dfs1(dep+1,s-c[dep],u+pow(2,dep-1));
}

void dfs2(int dep,int s,int u)
{
    if (s!=0) fl=true;
    if (fl&&dep>n)
    {
        p2++;
        b[p2].o=s;
        b[p2].q=u;
        if (s==0&&!v[u]) 
        {
            ans++;
            v[u]=true;
        }
        return;
    }
    if (dep>n) return;
    dfs2(dep+1,s,u);
    dfs2(dep+1,s+c[dep],u+pow(2,dep-1));
    dfs2(dep+1,s-c[dep],u+pow(2,dep-1));
}

int so(arr2 x,arr2 y)
{
    return x.o>y.o;
}

int so1(arr1 x,arr1 y)
{
    return x.o<y.o;
}

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
        scanf("%d",&c[i]);
    dfs1(1,0,0);
    fl=false;
    dfs2(n/2+1,0,0);
    sort(a+1,a+p1+1,so1);
    sort(b+1,b+p2+1,so);
    int i,j;
    i=j=1;
    while (i<=p1&&j<=p2)
    {
        if (a[i].o+b[j].o>0) j++;
            else if (a[i].o+b[j].o<0) i++;
        if (a[i].o+b[j].o==0)
        {
            int ii=i,jj=j;
            while (a[ii].o==a[ii].o)
            {
                int o1=a[ii].q+b[jj].q;
                if (!v[o1])
                {
                    v[o1]=true;
                    ans++;
                }
                if (b[jj].o==b[jj+1].o) jj++;
                    else if (a[ii].o==a[ii+1].o)
                         {
                            ii++;
                            jj=j;
                         }
                    else break; 
            }
            i=ii+1;
            j=jj+1;
        }
    }
    printf("%lld",ans);
}