本文最后更新于:3 years ago
题目链接:点击跳转
题意:有n个西瓜,每个可以买整个和买半个,问如果他想要购买西瓜的重量和分别为k=1,2,3…M时,有多少种购买西瓜的方案。
思路:将每个西瓜和已经买下的使用dp进行组合,注意半个和整个要独立存入
代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
| #include<bits/stdc++.h> using namespace std; typedef long long ll;
const ll MOD = 1e9 + 7; const int N = 1e3 + 5e2;
signed main() { ios::sync_with_stdio(false); cin.tie(nullptr);cout.tie(nullptr); int n, m; cin >> n >> m; vector<int>res(N << 1, 0); for (int i = 0; i < n; i++) { int a; cin >> a; res[0] = 1; vector<int>v = res, q = res; v[0] = q[0] = 1; for (int j = 0; j <= m; j++) { v[j + a] = (res[j + a] + res[j]) % MOD; } for (int j = 0; j <= m; j++) { q[j + a / 2] = (res[j + a / 2] + res[j]) % MOD; } for (int j = 1; j <= m; j++) { res[j] = (v[j] + q[j] - res[j] + MOD) % MOD; } } for (int i = 1; i <= m; i++) { if (i != 1) cout << ' '; cout << res[i]; } cout << endl; return 0; }
|