蒜头君去书店买书,他有 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
dfs深度搜索 当总价格和总数量与初始化相等时 可以购买 否则不能购买 略微剪枝;
ac:
#include <bits/stdc++.h>
using namespace std;
#define long long long
#define INF 0x3f3f3f3f
#define MOD 1000000007
#define MAX_N 1000010
typedef pair<int, int> PII;
typedef pair<long, long> PLL;
typedef pair<char, int> PCI;
long m, n, k, a[110];
void dfs(long p, long sum, long num)
{
if(num == k && sum == m)
{
cout << "Yes" << endl;
exit(0);
}
if(num > k || sum > m || p > n)
return ;
for(int i = 1; i <= 2; i++)
{
if(i & 1)
{
dfs(p + 1, sum + a[p], num + 1);
}
else
{
dfs(p + 1, sum, num);
}
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
#ifdef DEBUG
freopen("data.txt", "r", stdin);
#endif // DEBUG
cin >> m >> n >> k;
for(int i = 1; i <= n; i++)
{
cin >> a[i];
}
dfs(1, 0, 0);
cout << "No" << endl;
return 0;
}