洛谷P1064 金明的預算方案 dp 01揹包
阿新 • • 發佈:2018-12-23
題解
帶有限制的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;
}