《挑战程序竞赛》 2.1.4 部分和问题

题意：给定整数a1,a2,a3,...,an，判断是否可以从中选出若干数，使它们的和恰好为k。

 

解法：利用dfs深度优先遍历，从a1开始按顺序决定每个数是加还是不加。

 

code：

#include <iostream>
#include <algorithm>
using namespace std;

#define maxn 100000

int k, n;

int a[maxn];

bool dfs(int i, int sum)
{
    if(i == n)
        return sum == k;

    if(dfs(i + 1, sum + a[i]))
        return true;

    if(dfs(i + 1, sum))
        return true;

    return false;

}

int main()
{

    while(cin>>n)
    {
        for(int i=0; i<n; i++)
            cin>>a[i];
        cin>>k;
        if(dfs(0, 0))
            cout<<"Yes"<<endl;
        else
            cout<<"No"<<endl;

    }
    return 0;
}