Baby 今天清点自己的百宝箱啦,箱子里有 n 种硬币,硬币的面值分别是:val[1],
val[2],…,val[n],每种面值的硬币都恰好有 2 个。Baby 实在闲的太无聊了,
他想从他所拥有的硬币中选出若干个,使得面值之和为 k。那么他的目标能否实现
呢
输入
先输入 T,代表数据一共有 T 组,(T ≤ 100)每一组数据第一行都包含两个数字 n(n≤18),k(1≤k≤10^9)。n 代表箱子中所包含的硬币种数,k 代表 Baby 需要组成的金钱数额。接下来的一行代表 val[1],val[2],…,val[n]。(1≤val[i]≤ 10^7)
输出
如果Baby能组成金钱数额k,请输出Yes,否则输出No。
样例输入
2
2 10
3 4
3 9
1 2 10
样例输出
Case 1: Yes
Case 2: No
题解
二分折半查询。每种硬币有三种选择:选择其中一个,选择其中两个,不选择。
因此共有 3^18 种,对总的方案折半,以其中一半为基础,对另一半二分查询是
否存在可能使得构成 k。
#include <stdio.h>
#include <set>
std::set<int> st1, st2;
int a[20];
void dfs(int R,int id,int sum,std::set<int>&st){
if (id == R){
st.insert(sum);
return;
}
dfs(R,id+1,sum,st);
dfs(R,id+1,sum+a[id],st);
dfs(R,id+1,sum+2*a[id],st);
}
bool check(int K){
for(std::set<int>::iterator it = st1.begin(); it != st1.end(); it ++){
if(st2.find(K-*it) == st2.end()) continue;
return true;
}
return false;
}
int main(){
// freopen("baby_coins.in","r", stdin); //打开"baby_coins.in"文件,把里面的内容作为标准输入
// freopen("baby_coins.out","w", stdout);
int T, n, K;
int ica = 1;
scanf("%d", &T);
while(T --){
scanf("%d%d", &n, &K);
for(int i = 0; i < n; i ++){
scanf("%d", &a[i]);
}
int ok = false;
if(n == 1){
if(a[0] == K || a[0]*2 == K) ok = 1;
}
else{
dfs(n/2, 0, 0, st1);
dfs(n, n/2, 0, st2);
ok = check(K);
}
printf("Case %d: %s\n", ica ++, ok ? "Yes":"No");
st1.clear();
st2.clear();
}
return 0;
}