寻找和为定值的若干个数
给定整数a1、a2、a3、...、an,判断是否可以从中选出若干个数,使得它们的和等于k(k任意给定,且满足-10^8 <= k <= 10^8)。
分析:此题相对于本节“寻找满足条件的多个数”如出一辙,不同的是此题只要求判断,不要求把所有可能的组合给输出来。因为此题需要考虑到加上a[i]和不加上a[i]的情况,故可以采用深度优先搜索的办法,递归解决。
#define SIZE 10 #define SUM 20 list<int> mylist; void sumOfkNum(int sum, int*arr, int i, bool &flag) { if (sum <= 0 || i == SIZE || flag) { return; } if (sum == arr[i]) { flag = true; /*for (list<int>::iterator it = mylist.begin(); it != mylist.end(); ++it) { cout << *it << "+"; } cout << arr[i] << endl;*/ } mylist.push_back(arr[i]); sumOfkNum(sum - arr[i], arr, i + 1, flag); mylist.pop_back(); sumOfkNum(sum, arr, i + 1, flag); }
看到一个帖子的代码:
http://www.2cto.com/kf/201406/309134.html
简单的深搜代码:
#include <stdio.h> int n, k, ok, arr[22], vis[22], count; void DFS(int pos){ if(count >= k){ if(count == k){ if(!ok){ ok = 1; printf("YES\n"); } for(int i = 0; i < n; ++i) if(vis[i]) printf("%d ", arr[i]); printf("\n"); } return; } for(int i = pos; i < n; ++i){ count += arr[i]; vis[i] = 1; DFS(i + 1); count -= arr[i]; vis[i] = 0; } } int main(){ while(scanf("%d%d", &n, &k) == 2){ ok = 0; for(int i = 0; i < n; ++i){ scanf("%d", arr + i); vis[i] = 0; } count = 0; DFS(0); if(!ok) printf("NO\n"); } return 0; }
每天明白一点知识