求和问题(DFS)

输入:

n=4

a={1,2,4,7}

k=13

输出:

YES

输入:

n=4

a={1,2,4,7}

k=13

输出:

NO

选出若干数,使它们的和恰好为k

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <iostream>
#include <ctype.h>
#include <iomanip>
#include <queue>
#include <stdlib.h>
using namespace std;
 
int a[55];
int n,k;
//已经从前i项得到了和sum,然后对于i项之后的进行分支
bool dfs(int i,int sum)
{
    //如果前n项都计算过了,则返回sum是否与k相等
    if(i==n)  return sum==k;
    //不加a[i]的情况
    if(dfs(i+1,sum))  return true;
    //加上a[i]的情况
    if(dfs(i+1,sum+a[i]))  return true;
    //无论是否加上a[i]都不能凑成k就返回false
    return false;
}

void solve()
{
    if(dfs(0,0)) printf("YES
");
    else   printf("NO
");
}

int main()
{
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    cin>>k;
    solve();
}

<<挑战程序设计竞赛>>读后感