51nod1268 和为K的组合 折半枚举

来源:互联网 发布:php exec 返回值126 编辑:程序博客网 时间:2024/05/16 13:55

题目:

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1268

题意:

给出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”。

思路:

折半枚举预处理,然后二分查找即可

#include <bits/stdc++.h>using namespace std;const int N = 10000 + 10;int a[N], b[N];int arr[N];int work(int *a, int *arr, int m, int len){    int k = 0;    for(int i = 0; i < (1<<m); i++)    {        ++k;        for(int j = 0; j < m; j++)            if((i>>j) & 1) a[k] += arr[len-j];    }    return k;}int main(){    int n, k;    while(~ scanf("%d%d", &n, &k))    {        for(int i = 1; i <= n; i++) scanf("%d", &arr[i]);        memset(a, 0, sizeof a);        memset(b, 0, sizeof b);        int ka = work(a, arr, n / 2, n / 2);        int kb = work(b, arr, n - n/2, n);        sort(a + 1, a + 1 + ka);        sort(b + 1, b + 1 + kb);        bool flag = false;        for(int i = 1; i <= ka; i++)        {            if(binary_search(b + 1, b + 1 + kb, k - a[i]))            {                flag = true; break;            }        }        puts(flag ? "Yes" : "No");    }    return 0;}
原创粉丝点击