51nod 1268

1268 和为K的组合 

基准时间限制：1 秒 空间限制：131072 KB 分值: 20 难度：3级算法题

 收藏

 关注

给出N个正整数组成的数组A，求能否从中选出若干个，使他们的和为K。如果可以，输出："Yes"，否则输出"No"。

Input

第1行：2个数N, K, N为数组的长度, K为需要判断的和(2 <= N <= 20，1 <= K <= 10^9)
第2 - N + 1行：每行1个数，对应数组的元素A[i] (1 <= A[i] <= 10^6)

Output

如果可以，输出："Yes"，否则输出"No"。

Input示例

5 13
2
4
6
8
10

Output示例

No

数量小，可以直接做，用一集合储存现在的和，每遍历一个数就让它和集合里的数都相加。当然，只有小于k的才放进集合里，这个可以少算一些。

#include <bits/stdc++.h>
using namespace std;
int a[22];
int main() {
    int n, k;
    set<int> st[2];
    set<int> ::iterator it;
    cin >> n >> k;
    int last = 0;
    for(int i = 1; i <= n; i ++) cin >> a[i];
    for(int i = 1; i <= n; i ++) {
        last = 1 - last;
        it = st[1-last].begin();
        for(; it != st[1-last].end(); ++it) {
            if((*it)+a[i] < k) {
                st[last].insert((*it)+a[i]);
            } else if((*it)+a[i] == k) return 0*printf("Yes
");
            st[last].insert((*it));
        }
        st[last].insert(a[i]);
    }
    if(st[last].count(k)) printf("Yes
");
    else printf("No
");
    return 0;
}