目录
一、题目
1、题目描述
2、输入输出
2.1输入
2.2输出
3、原题链接
|-----------------------------------------------------------------------------------------------------------------------------------------|
| G2 - Playlist for Polycarp (hard version) |
二、解题报告
1、思路分析
一眼dp，但是这个dp思路和代码都不太好想
首先是涉及到组合数学的分配问题
方案数怎么求？
既涉及到了相邻的顺序，又涉及到了容量/费用
我们可以单独考虑然后相乘：
f(i, j, k, x) 为 选取 i 个类型1，j个 类型2，k 个类型3，并且以类型x结尾，且无相同相邻项的方案数
这个dp可太简单了，O(1)转移，O(N^3 * 3)的方案数，很好搞定
再考虑 g(i, j, k, t) 即 选 i 个类型1，j个 类型2，k 个类型3，总费用为t（费用指的是时间和）的方案数
这是个费用背包问题，我们直接求是O(N ^ 4 T)，太大了
考虑转换一下，g(j, k, p) 为 j个 类型2，k 个类型3 总费用为t方案数，h(i, T - p)为i个类型1，总费用为t的方案数，乘法原理二者相乘可得
然后 (f(i, j, k, 0) + f(i, j, k, 1) f(i, j, k, 2)) * g(j, k, p) * h(i, T - p) * fac(i) * fac(j) * fac(k) 就是一组方案数
什么意思？
f 确定了每个类型放的位置，然后每个类型的每个物品是不同的，这就是一个多重集排列问题
然后 h 和 t 又确定了选取哪些类型1、2、3，再根据乘法原理乘一块就是答案
2、复杂度
时间复杂度： O(N^3 T)空间复杂度：O(N^ T)
3、代码详解
cpp
#include <bits/stdc++.h>
#define sc scanf
using i64 = long long;
using PII = std::pair<int, int>;
constexpr int N = 55, M = 2505, P = 1'000'000'007;
void add(int& x, int y) { x += y, (x >= P) && (x -= P); }
int fac[N], f[N][N / 2][N / 3][3], g[N / 2][N / 3][M], h[N][M];
void solve() {
int n, T;
std::cin >> n >> T;
fac[0] = 1;
for (int i = 1; i <= n; ++ i) fac[i] = 1LL * i * fac[i - 1] % P;
std::vector<int> a, b, c;
std::vector<std::array<int, 3>> cnt(n);
for (int i = 0, t, g; i < n; ++ i) {
std::cin >> t >> g;
if (g == 1) a.push_back(t);
if (g == 2) b.push_back(t);
if (g == 3) c.push_back(t);
}
if (a.size() < b.size())
std::swap(a, b);
if (a.size() < c.size())
std::swap(a, c);
if (b.size() < c.size())
std::swap(b, c);
int A = a.size(), B = b.size(), C = c.size();
// 求费用背包
g[0][0][0] = 1;
for (int i = 0; i < B; ++ i)
for (int j = i; ~j; -- j)
for (int p = T - b[i]; p >= 0; -- p)
add(g[j + 1][0][p + b[i]], g[j][0][p]);
for (int i = 0; i < C; ++ i)
for (int j = B; j >= 0; -- j)
for (int k = i; k >= 0; -- k)
for (int p = T - c[i]; p >= 0; -- p)
add(g[j][k + 1][p + c[i]], g[j][k][p]);
h[0][0] = 1;
for (int i = 0; i < A; ++ i)
for (int j = i; ~j; -- j)
for (int p = T - a[i]; p >= 0; -- p)
add(h[j + 1][p + a[i]], h[j][p]);
// 刷表
f[1][0][0][0] = f[0][1][0][1] = f[0][0][1][2] = 1;
for (int i = 0; i <= A; ++ i)
for (int j = 0; j <= B; ++ j)
for (int k = 0; k <= C; ++ k) {
for (int x = 0, v; x < 3; ++ x) {
if (v = f[i][j][k][x]) {
if (x) add(f[i + 1][j][k][0], v);
if (x ^ 1) add(f[i][j + 1][k][1], v);
if (x ^ 2) add(f[i][j][k + 1][2], v);
}
}
add(f[i][j][k][0], f[i][j][k][1]);
add(f[i][j][k][0], f[i][j][k][2]);
f[i][j][k][0] = 1LL * f[i][j][k][0] * fac[i] % P * fac[j] % P * fac[k] % P;
}
int res = 0;
for (int j = 0; j <= B; ++ j)
for (int k = 0; k <= C; ++ k)
for (int p = 0; p <= T; ++ p)
if (g[j][k][p])
for (int i = 0; i <= A; ++ i)
if (h[i][T - p])
add(res, 1LL * f[i][j][k][0] * g[j][k][p] % P * h[i][T - p] % P);
std::cout << res;
}
int main() {
#ifdef DEBUG
freopen("in.txt", "r", stdin);
freopen("out.txt", "w", stdout);
#endif
std::ios::sync_with_stdio(false), std::cin.tie(nullptr), std::cout.tie(nullptr);
int _ = 1;
// std::cin >> _;
while (_ --)
solve();
return 0;
}