【模版】背包问题汇总

【模版】背包问题汇总

写在前面

多重背包优化方案还有单调队列优化(暂未实现)
单调队列优化参考网址：https://www.cnblogs.com/-guz/p/9866118.html
0-1背包第二层循环为什么倒序，完全背包第二层循环为什么正序

01背包模版

//01背包模版
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define MaxSize 100010  //注意最大物品数的修改
int wei[MaxSize], val[MaxSize], dp[MaxSize]; //物品重量 价值 最优答案
int main() {
    int n, cap; //物品数量number 容量capacity
    scanf("%d%d", &n, &cap);
    for (int i = 1; i <= n; i++) scanf("%d%d", &wei[i], &val[i]);
    for (int i = 1; i <= n; i++)
        for (int j = cap; j >= wei[i]; j--)
            dp[j] = max(dp[j], dp[j-wei[i]]+val[i]);
    printf("%d\n", dp[cap]);
    return 0;
}

完全背包模版

//完全背包模版
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define MaxSize 100010  //注意最大物品数的修改
int wei[MaxSize], val[MaxSize], dp[MaxSize]; //物品重量 价值 最优答案
int main() {
    int n, cap; //物品数量number 容量capacity
    scanf("%d%d", &n, &cap);
    for (int i = 1; i <= n; i++) scanf("%d%d", &wei[i], &val[i]);
    for (int i = 1; i <= n; i++)
        for (int j = wei[i]; j <= cap; j++)
            dp[j] = max(dp[j], dp[j-wei[i]]+val[i]);
    printf("%d\n", dp[cap]);
    return 0;
}

多重背包模版【二进制优化】

//多重背包模版(二进制优化)
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define MaxSize 100010  //注意最大物品数的修改
int wei[MaxSize], val[MaxSize], num[MaxSize], dp[MaxSize]; //物品重量 价值 数量 最优答案
int nwei[MaxSize<<4], nval[MaxSize<<4];   //  新的物品重量 物品价值 开足够大的空间
int main() {
    int n, cap; //物品数量number 容量capacity
    scanf("%d%d", &n, &cap);
    int pos = 0;    //新数组定位 最终值为新合成物品总数量
    for (int i = 1; i <= n; i++) scanf("%d%d%d", &wei[i], &val[i], &num[i]);
    for (int i = 1; i <= n; i++){
        for (int j = 1; j <= num[i]; j<<=1){
            num[i] -= j;    //减去拆分数量
            nwei[++pos] = j*wei[i]; //存放合成物品体积
            nval[pos] = j*val[i];   //存放合成物品价值
        }
        if (num[i]){    //如有剩余部分
            nwei[++pos] = num[i]*wei[i];
            nval[pos] = num[i]*val[i];
            num[i] = 0; //数量拆分完毕
        }
    }
    for (int i = 1; i <= pos; i++)  //注意这里是新的物品量
        for (int j = cap; j >= nwei[i]; j--)
            dp[j] = max(dp[j], dp[j-nwei[i]]+nval[i]);
    printf("%d\n", dp[cap]);
    return 0;
}