【DFS】DLUToj-1215-组合数字

题目链接：http://acm.dlut.edu.cn/problem.php?id=1215

题目描述：给出N个整数，每个数范围在[-1000,1000]，问能不能在N个数中选出一些数（但不可以不选，且每个数最多只能被选择一次），使他们的和为0

解题思路：OJ上竟然有人发邮件问我。。点开一看还是我没做过的题。。一道DFS

好吧。。去敲了一下。显然得剪枝一下才能过，懒得剪枝WA了一下，改完就过了

把数字分成正数和负数
sum大于0从负数那边找
sum小于0从正数那边找

竟然还跑了0ms。。。无语

AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

int arr[50],vis[50];
int n,sum,found,mid,fu,zh;

void dfs(int i)
{
    //cout<<sum<<endl;
    if(found)return;
    if(!sum)
        found=1;
    else
    {
        vis[i]=1;
        int j;
        if(sum>0)
        {
            for(j=fu+1;j<mid;j++)
            {
                if(!vis[j])
                {
                    vis[j]=1;
                    sum+=arr[j];
                    fu++;
                    dfs(j);
                    fu--;
                    vis[j]=0;
                    sum-=arr[j];
                }
            }
        }
        else
        {
            for(j=zh++;j<n;j++)
            {
                if(!vis[j])
                {
                    vis[j]=1;
                    sum+=arr[j];
                    dfs(j);
                    vis[j]=0;
                    sum-=arr[j];
                }
            }
        }
    }
}

int solve()
{
    int i;
    for(i=0;i<n;i++)
    {
        if(arr[i]>0)return 0;
        sum=arr[i];
        found=0;
        fu=i;zh=mid;
        memset(vis,0,sizeof(vis));
        dfs(i);
        if(found)
            return 1;
        return 0;
    }
}

int main()
{
    freopen("input.txt","r",stdin);
    int T,i;
    scanf("%d",&T);
    while(T--)
    {
        memset(arr,0,sizeof(arr));
        scanf("%d",&n);
        for(i=0;i<n;i++)
            scanf("%d",&arr[i]);
        sort(arr,arr+n);
        for(i=0;i<n;i++)
        {
            if(arr[i]>0)
            {
                mid=i;break;
            }
        }
        int ans=solve();
        if(ans)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}