部分和问题 --dfs

部分和问题
给定整数 a1、a2、…、an,判断是否可以从中选出若干数,使它们的和恰好为 k。 限制条件  1 ≤ n ≤ 20 , 108 ≤ ai ≤ 108  108 ≤ k ≤ 108
样例输入 n=4 a={1,2,4,7} k=13
输出 Yes (13 = 2 + 4 + 7)

分析:每个数都有选与被选两种状态,一开始需要有一个初始状态(全选或者全不选),我这里是一开始全都不选(这种情况是肯定凉的),然后从a1开始遍历一直到an,判断其和是不是为k,是的话就输出。然后再改变状态回到上一步继续开始遍历。

 

#include <iostream>
using namespace std;
int n,a[20],k;//数组b是用来标识状态的,b[i]为0则代表不选,为1则是选
bool dfs(int step,int sum){//step表示的当前的位置 
    if(step==n)        return sum==k;//如果相等则返回一,这里应该与n比较而非n-1,因为相加的操作是通过递推到下一步实现的 
    //只要最终step==n时为true,,则一趟返回下来全部为true
    if(dfs(step+1,sum))    return true; //这个是当前位置处不加,也就是该函数起始是什么都不加的遍历到最后 
    if(dfs(step+1,sum+a[step]))     return true;//这个是加的情况
    return false;//最后一定要加false保证循环终止 
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    cin>>k;
    if(dfs(0,0))    cout<<"YES\n";
    else cout<<"NO\n"; 
    return 0;
}

 

posted @ 2018-07-24 15:28  清酒令  阅读(397)  评论(0编辑  收藏  举报