《挑战程序设计竞赛》例题:部分和问题

例题大意:给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";
}
全部评论

相关推荐

我见java多妩媚:大外包
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务