蒜头君去书店买书,他有 mm 元钱,书店里面有 nn 本书,每本书的价格为 p_ipi 元。蒜头君很爱学习,想把身上钱都用来买书,并且刚好买 kk 本书。请帮蒜头君计算他是否能刚好用 mm 元买 kk本书。
输入格式
第一行输入 33 个整数 m(1 \le m \le 100000000)m(1≤m≤100000000),n(1 \le n \le 30)n(1≤n≤30),k(1 \le k \le min(8, n))k(1≤k≤min(8,n))
接下来一行输入 nn 个整数,表示每本书的价格 p_i(1 \le p_i \le 100000000)pi(1≤pi≤100000000)。
输出格式
如果蒜头君能 刚好 用 mm 元买 kk 本书,输入一行"Yes"
, 否则输出"No"
。
样例输入1
10 4 4 1 2 3 4
样例输出1
Yes
样例输入2
10 4 3 1 2 3 4
样例输出2
No
解题思路:深搜+剪枝 每一本书都有两种状态 即买或者不买 如果买该数 则sum=sum+该书的价格 买的书的数量++ 即ans++
不买则跳过
剪枝 : 当 买的书的数量和价格都合适时 输出“Yes” 当价格大于给定价格时 return 当买的书的数量>给定数量是return
AC代码:
#include <bits/stdc++.h>
#include <vector>
using namespace std;
int m,n,k,sum=0,ans=0;
vector<int> a;
void dfs(int key)
{
if(sum==m&&ans==k)
{
cout<<"Yes";
exit(0);
}
if(ans==k)
return ;
if(key==n)
return ;
for(int i=1;i<=2;i++)
{
if(i%2&&a[key]+sum<=m)//1 代表取该项 2代表不取
{
sum+=a[key];
ans++;
dfs(key+1);
ans--;
sum-=a[key];
}
else
dfs(key+1);
}
return ;
}
int main()
{
cin>>m>>n>>k;
for(int i=0;i<n;i++)
{
int tmp;
cin>>tmp;
a.push_back(tmp);
}
dfs(0);
cout<<"No";
return 0;
}