題解

帶有限制的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;
}