寻找和为定值的若干个数

给定整数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;
}
View Code

 

posted @ 2017-06-21 17:03  泡面小王子  阅读(754)  评论(0编辑  收藏  举报