当前位置: 代码迷 >> 综合 >> NYOJ - 1058 - 部分和问题(DFS+减枝)
  详细解决方案

NYOJ - 1058 - 部分和问题(DFS+减枝)

热度:74   发布时间:2023-10-09 18:07:17.0
描述
给定整数a1、a2、.......an,判断是否可以从中选出若干数,使它们的和恰好为K。
输入
首先,n和k,n表示数的个数,k表示数的和。
接着一行n个数。
(1<=n<=20,保证不超int范围)
输出
如果和恰好可以为k,输出“YES”,并按输入顺序依次输出是由哪几个数的和组成,否则“NO”
样例输入
4 13
1 2 4 7

样例输出

YES
2 4 7

题目思路:拿到这道题,看到从给定数列中选出若干个数,是它们的和恰好为K的时候,首先想到子集,用子集生成的递归算法,求出所有子集,判断符合条件的子集并且输出。后来想了半天,不知道怎么写。于是改变了思路。用深度搜索,从左往右,挨个元素搜索,但是递归的时候只搜索当前元素右边的元素。(其实就是求全排列的方法,只不过不搜左边,因为相同数字,顺序不同则是不同的排列,但却是相同的子集)。递归的边界是sum>=k,搜到以后直接停止搜索。


#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int n,k;
int sum;
int a[22];
int vis[22];
int b;
void dfs(int cur){if(sum>=k){if(sum==k){b = 1;//用来判断是否存在有效输出 puts("YES");for(int i=0 ;i<n ;i++){if(vis[i])printf("%d ",a[i]);}puts("");}return ;	}for(int i=cur ;i<n ;i++){sum += a[i];vis[i] = 1;//设计使用标记 dfs(i+1);//这里的变量是i+1 不是cur+1 因为不搜索左边 sum -= a[i];vis[i] = 0;	//清除使用标记 }} int main(){while(scanf("%d%d",&n,&k)!=EOF){sum = 0;b=0;memset(vis,0,sizeof(vis));for(int i=0 ;i<n ;i++){scanf("%d",&a[i]);}dfs(0);if(!b)puts("NO");}return 0;
}


??