描述
给定整数a1、a2、…an,判断是否可以从中选出若干数,使它们的和恰好为K。
输入
首先,n和k,n表示数的个数,k表示数的和。
接着一行n个数。
(1<=n<=20,保证不超int范围)
输出
如果和恰好可以为k,输出“YES”,并按输入顺序依次输出是由哪几个数的和组成,否则“NO”
样例输入
4 13
1 2 4 7
样例输出
YES
本题就是利用深搜法从最第一个数开始判断后面的数加或不加,直到满足几个数相加等于K时结束。
当时找别人这一题代码详解找了半天都没找到,直到自己看懂了才发现自己也描述不清楚。
#include<cstdio>
#include<algorithm>
using namespace std;
int a[1000];
int n, k;
//已知前i项中的任意几项的和sum,对i后的几项进行分支
bool dfs(int i, int sum)
{
if (i == n)//如果前n项都计算过了,判断是否与所给的k值相等
return sum == k;
if (dfs(i + 1, sum))//如果不加上第a[i]项和为K
return true;
if (dfs(i + 1, sum + a[i]))//如果加上第a[i]项和为K
return true;
return false;//如果这几种情况都不能凑成和为K
}
void solve()
{
if (dfs(0, 0))
printf("YES\n");
else
printf("NO\n");
}
int main()
{
int i;
scanf("%d%d", &n, &k);
for (i = 0; i < n; i++)
scanf("%d", &a[i]);
solve();
}