装背包的方案数

欢迎访问我的博客首页。

---

1. 背包问题方案数

---

  简单背包问题只考虑体积。
  题目来自 牛客网：小 Q 有 X 首长度为 A 的不同的歌和 Y 首长度为 B 的不同的歌，现在小 Q 想用这些歌组成一个总长度正好为 K 的歌单，每首歌最多只能在歌单中出现一次，在不考虑歌单内歌曲的先后顺序的情况下，请问有多少种组成歌单的方法。

#include<iostream>
#include<vector>
using namespace std;

int simple_package(vector<int>& volumes, int capacity) {
	if (volumes.size() == 0 || capacity <= 0)
		return 0;
	const int mod = 1000000007;

	vector<vector<int>> dp(capacity + 1, vector<int>(volumes.size()));
	// 1.计算dp[0][0]：把体积为0的物体放入容积为0的容器中的放法为1。
	dp[0][0] = 1;
	// 2.计算dp[1:][0]：把体积为volumes[0]的物体放入容积为volumes[0]的容器中的放法为1。
	for (int i = 1; i <= capacity; i++)
		dp[i][0] = i == volumes[0] ? 1 : 0;
	// 3.计算dp[:][1:]。
	for (int i = 0; i <= capacity; i++)
		for (int j = 1; j < volumes.size(); j++) {
			if (volumes[j] <= i)
				dp[i][j] = (dp[i - volumes[j]][j - 1] + dp[i][j - 1]) % mod;
			else
				dp[i][j] = dp[i][j - 1];
		}
	return dp[capacity][volumes.size() - 1];
}
// dp[i][j]=x：把volumes[0:j]这j+1个物体放入容积为i的容器中的方法共有x种。

int main() {
	int simple_package(vector<int>& volumes, int capacity);
	int a, x, b, y, k;
	cin >> k >> a >> x >> b >> y;
	vector<int> volumes;
	for (int i = 1; i <= x; i++)
		volumes.push_back(a);
	for (int i = x + 1; i <= x + y; i++)
		volumes.push_back(b);
	cout << simple_package(volumes, k) << endl;
}