题解
带有限制的01背包 最开始以为是状压dp发现状态太多
按照01背包的写法 将主件和附件看作同一个物品 处理时计算只买主件、买主件和第一个附件、买主件和第二个附件、买主件和两个附件四种情况进行转移 转移枚举容量时这四种情况并行操作就不会相互造成贡献 不同物品之间则可以
接受数据时直接将权值*价格方便计算
AC代码
#include <stdio.h>
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const int MAXN = 100;
const int MAXM = 4e4;
struct node
{
int v, w; //重量 价值
};
vector<node> a[MAXN];
int d[MAXM];
int main()
{
#ifdef LOCAL
freopen("C:/input.txt", "r", stdin);
#endif
int N, M; //物品 容量
cin >> M >> N;
for (int i = 1; i <= N; i++)
{
int v, p, q;
scanf("%d%d%d", &v, &p, &q);
p *= v; //直接计算价值
if (q)
a[q].push_back({ v, p });
else
a[i].push_back({ v, p });
}
for (int i = 1; i <= N; i++)
if (a[i].size())
for (int j = M; j >= 0; j--)
{
node f = a[i][0];
if (j >= f.v) //只买主件
d[j] = max(d[j], d[j - f.v] + f.w);
if (a[i].size() > 1)
{
node s1 = a[i][1];
if (j >= f.v + s1.v) //主+1附
d[j] = max(d[j], d[j - f.v - s1.v] + f.w + s1.w);
if (a[i].size() > 2)
{
node s2 = a[i][2];
if (j >= f.v + s2.v) //主+2附
d[j] = max(d[j], d[j - f.v - s2.v] + f.w + s2.w);
if (j >= f.v + s1.v + s2.v) //主+12附
d[j] = max(d[j], d[j - f.v - s1.v - s2.v] + f.w + s1.w + s2.w);
}
}
}
cout << d[M] << endl;
return 0;
}