題目:
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;
}