Description
Baby 今天清点自己的百宝箱啦,箱子里有 n 种硬币,硬币的面值分别是:val[1],val[2],…,val[n],每种面值的硬币都恰好有 2 个。Baby 实在闲的太无聊了,他想从他所拥有的硬币中选出若干个,使得面值之和为 k。那么他的目标能否实现呢 ~
Input
每一组数据第一行都包含两个数字 n(n≤18),k(1≤k≤109)。n 代表箱子中所包含的硬币种数,k 代表 Baby 需要组成的金钱数额。接下来的一行代表 val[1],val[2],…,val[n]。(1≤val[i]≤ 107)
Output
如果Baby能组成金钱数额k,请输出Yes,否则输出No。
Sample Input
2
2 10
3 4
3 9
1 2 10
Sample Output
Case 1: Yes
Case 2: No
#include <stdio.h>
#include <set>
std::set<int>st1,st2;//程序采用二分查找,前半部分为st1,后半部分为st2;
int a[20];
void dfs(int l, int r,int sum,std::set <int> &st){
if (l == r){
st.insert(sum); //用bfs寻找前、后半部分可能的和分别计入st1,st2;
return;
}
dfs(l+1, r, sum,st);
dfs(l+1, r, sum+a[l],st);
dfs(l+1, r, sum+2*a[l],st);
}
int check(int k){
for(std::set<int> ::iterator it = st1.begin(); it != st1.end(); it ++){
if(st2.find(k-*it) == st2.end()) continue;//在st2中寻找满足st2=k-st1的元素,如果找到,返回1,否则,继续寻找,直到不在范围之内
return 1;
}
return 0;
}
int main(){
int t,q=1;
scanf("%d",&t);
while(t--){
int n,k;
scanf("%d %d",&n,&k);
for(int i=0;i<n;i++){
scanf("%d", &a[i]);
}
int ok = 0;
st1.clear();
st2.clear();//清除st1,st2
// sort(a,a+n);
if(n == 1){
if(a[0] == k || a[0] *2 == k){
ok=1;
}
}
else{
dfs(0,n/2,0,st1);
dfs(n/2,n,0,st2);
ok = check(k);
}
printf("Case %d: %s\n", q++, ok ? "Yes":"No");
}
}