xynuoj 1277: 子集和问题（dfs）

1277: 子集和问题

时间限制: 1 Sec   内存限制: 128 MB
提交: 30   解决: 8
您该题的状态：已完成
[提交][状态][讨论版]

题目描述

【问题描述】对于一个给定正整数的集合s={x1,x2,x3…xn}和正整数c，编程计算s的一个子集s1，使得子集s1的和等于c。

【输入格式】

第一行有2个正整数n和c，第二行有n个正整数

【输出格式】 

一行数据，按输入的顺序输出，若无解则输出"No Solution!"

【输入样例】

5 10

2 2 6 5 4

【输出样例】

2 2 6

【问题规模】

n<7000,c<maxlongint

/*这道题和整数凑和差不多，只不过没要求要选择一定量的元素，
但是应该注意当所有元素和都小于c时，
都不可能有解，此时如果调用dfs(),
要遍历所有组合才能得到无解结论，会超时
所以预判一下。 
*/ 
#include<stdio.h>
#include<string.h>
using namespace std;
long long mat[7009];
int vis[7009];
long long a[7009];
long long c;
int flag,n;
int dfs(long long sum,int k)
{
	if(sum>c) return 0;
    if(sum==c)
    {
        for(int i=0;i<k-1;i++)
            printf("%lld ",a[i]);
        printf("%lld\n",a[k-1]);
        return 1;
    }
    for(int i=0;i<n;i++){
        if(!vis[i])
        {
            vis[i]=1;
            a[k]=mat[i];
            if(dfs(sum+mat[i],k+1))//剪枝 ，如果有解直接return结束,无解才回溯 
            	return 1;
            vis[i]=0;
        }
    }
    return 0;
}
int main()
{
    scanf("%d%lld",&n,&c);
    long long sum=0;
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++)
        {
        	 scanf("%lld",&mat[i]);
        	 sum+=mat[i];
		}
		if(sum<c) printf("No Solution!\n");//预判，如果所有元素和都小于c,则无解 
		else{
			
			if(dfs(0,0)==0) printf("No Solution!\n");
		}
        
         
    return 0;
}