DFS时,有时早已明确知道从当前状态无论如何转移都不会存在解。这种情况下不再搜索而是直接跳过。这种方法被成为剪枝。
NYOJ1056 部分和问题 这道题需要剪掉的是:
1.从当前状态如何转移都不会存在解
2.当sum超过k时,也没必要继续搜索
代码如下:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int sum,n,v,k,a[22],b[22];
void dfs(int x)
{ if(sum>=k){ //使用了剪枝的方法,当sum超过k时,也没必要继续搜索,直接return
if(sum==k){
if(v){v=0;printf("YES\n");}
for(int i=0;i<n;i++)
if(b[i])printf("%d ",a[i]);
printf("\n");
}
return ;
}
for(int i=x;i<n;i++) //核心代码,利用递归进行深搜
{
sum+=a[i];
b[i]=1;
dfs(i+1);
sum-=a[i];
b[i]=0;
}
}
int main()
{
while(scanf("%d%d",&n,&k)!=EOF)
{
v=1; //做标记使用
memset(b,0,sizeof(b));
for(int i=0;i<n;i++)scanf("%d",&a[i]);
sum=0;
dfs(0);
if(v)printf("NO\n");
}
return 0;
}