蒜头君去书店买书,他有 m 元钱,书店里面有 n 本书,每本书的价格为 pi 元。蒜头君很爱学习,想把身上钱都用来买书,并且刚好买 kkk 本书。请帮蒜头君计算他是否能刚好用 m 元买 k 本书。
输入格式
第一行输入 3 个整数 m(1≤m≤100000000)m,n(1≤n≤30)n(1 \le n \le 30)n(1≤n≤30)
接下来一行输入 n 个整数,表示每本书的价格 pi(1≤pi≤100000000)p)。
输出格式
如果蒜头君能 刚好 用 m 元买k 本书,输入一行"Yes", 否则输出"No"。
样例输入1
10 4 4
1 2 3 4
样例输出1
Yes
样例输入2
10 4 3
1 2 3 4
样例输出2
No
思路在于:搜索一个二叉树,对于每一本书都有买和不买两个分支
AC代码:
#include<iostream>
#include<cstdio>
using namespace std;
bool ret=false;
int m,n,k;
int a[30];
void dfs(int key,int temp,int sum)
{
if(key>n-1||temp>k||sum>m)//剪枝,如果循环次数(key)大于数组a,或者买的书(temp)大于所需的总数,或者买书所花的钱(sum)大于自己的钱(m)则打破递归
{
return;
}
if(m==sum&&k==temp)
{
ret=true;
return;
}
if(key<n)
{
dfs(key+1,temp+1,sum+a[key]);//买该本书,遍历下一本是否购买
dfs(key+1,temp,sum);//不买该书,遍历下一本书是否购买
}
}
int main()
{
//freopen("1.txt","r",stdin);
cin>>m>>n>>k;
for(int i=0;i<n;i++)
cin>>a[i];
dfs(0,0,0);
if(ret)
cout<<"Yes";
else
cout<<"No";
return 0;
}