《挑战程序设计竞赛》例题:部分和问题
例题大意:给n个数,再给一个数h,问n个数中有没有一组数的和为h
分析:该题用深度优先搜索(DFS),给的每个数都有两种状态,一种是取到了该数,一种是没取到该数,不管取没取到,它的下标肯定会+1,取到了的话sum就与这个位置的数相加(也就是sum+a[i],注意不是a[i+1],因为数组是从0开始的,如果是a[i+1]那么当人的下标为1时,a[0]这个数没有进行取或者不取,而且当i+1=n的时候,a[i+1]会越界)当某组数的和为h或者当前位置(下标)为数组大小n时为递归结束条件。感觉就像某个人去
#include <bits/stdc++.h> using namespace std; int n,h; int a[1000]; int dfs(int i,int sum) { if(i==n)return sum==h; if(sum==h)return 1;//这一句个人觉得加了快一些,因为可以提前结束递归,不需要等到i=n的时候 if(dfs(i+1,sum))return 1;//这是没有取到下标为i的数,如果和为h,那么返回1 if(dfs(i+1,sum+a[i]))return 1;//这是取到下标为i的数,如果和为h,也返回1 return 0; } int main() { cin>>n>>h; for(int i=0;i<n;i++) { cin>>a[i]; } if(dfs(0,0)==1)cout<<"yes"; else cout<<"no"; }