蒜头君去书店买书,他有 m 元钱,书店里面有n 本书,每本书的价格为 pi 元。蒜头君很爱学习,想把身上钱都用来买书,并且刚好买 k 本书。请帮蒜头君计算他是否能刚好用m 元买k 本书。
输入格式
第一行输入 3 个整数m(1≤m≤100000000),n(1≤n≤30),k(1≤k≤min(8,n))
接下来一行输入 n 个整数,表示每本书的价格pi(1≤pi≤100000000)。
输出格式
如果蒜头君能 刚好 用 m 元买 kk 本书,输入一行"Yes"
, 否则输出"No"
。
样例输入1
10 4 4 1 2 3 4
样例输出1
Yes
样例输入2
10 4 3 1 2 3 4
样例输出2
No
解题说明:这里有满足的情况返回1,所有情况不满足才返回0。所以记录一个ret,赋初值为0,当有满足的条件修改其为1,然后每次函数返回ret。
代码:
#include<iostream>
#include<algorithm>
using namespace std;
int book[35];
int m,k,n;
int ret=0;
int dfs(int sum,int c,int i){
if(sum==m&&c==k)return ret=1;
if(i<n&&c<k){
if(sum+book[i]<=m){
dfs(sum+book[i],c+1,i+1);
dfs(sum,c,i+1);
}
else dfs(sum,c,i+1);
}
return ret;
}
int main(){
cin>>m>>n>>k;
int ip1;
for(int i=0;i<n;i++){
cin>>ip1;
book[i]=ip1;
}
if(dfs(0,0,0))cout<<"Yes";
else cout<<"No";
return 0;
}