求和问题(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[];
 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+,sum))  return true;
     //加上a[i]的情况
     if(dfs(i+,sum+a[i]))  return true;
     //无论是否加上a[i]都不能凑成k就返回false
     return false;
 }

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

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

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