题目
代码
#include <bits/stdc++.h>
using namespace std;
const int N = 1e4+10;
int cnt[5]; // 存兼容i等级及以下的孔的数目的桶
int l[N], p[N]; // l[i] i号珠子等级 p[i] i号珠子的上限
int w[N][8], f[N]; // w[i][j] i号珠子镶嵌j个的值 f[i] 孔数为i的最大值
int m; // 珠子种类
int main()
{
for(int i = 1; i <= 6; i++)
{
int n;
cin >> n;
for(int j = 1; j <= n; j++)
{
int t;
cin >> t;
cnt[t]++; // 统计每个等级的孔数量
}
}
for(int i = 3; i; i--)
cnt[i] += cnt[i+1]; // 累加,使得 cnt[i] 包含等级为 i 及以上的所有孔数量
cin >> m;
for(int i = 1; i <= m; i++)
{
cin >> l[i] >> p[i]; // 输入第 i 种珠子的等级和上限
for(int j = 1; j <= p[i]; j++)
cin >> w[i][j]; // 输入第 i 种珠子镶嵌 j 个时的价值
}
int ans = 0;
for(int i = 4; i; i--) // 从高往低开放孔的使用
{
for(int j = 1; j <= m; j++) // 枚举各组
{
if(l[j] == i) // 组等级要等于 i(不重复选取)
{
for(int v = cnt[i]; ~v; v--) // 枚举孔数目
{
for(int t = 0; t <= p[j]; t++) // 枚举选择当前组物品数量的决策
{
if(v - t >= 0) f[v] = max(f[v], f[v - t] + w[j][t]); // 更新 f[v] 为当前孔数的最大价值
ans = max(ans, f[v]); // 更新全局最优解
}
}
}
}
}
cout << ans; // 输出结果
}
注意(From TOTP)