2022 icpc杭州站 C. No Bug No Game - 背包dp

题面

分析

能拿整个 p i p_i pi的就拿整个的,不能拿了可以拿一部分的,因此可以分成0和1两种情况,0表示拿整个的,1表示还可以拿部分的,两种情况放在一起做一遍01背包,找到最大价值。

代码
cpp 复制代码
#include <bits/stdc++.h>

#define int long long

using namespace std;

const int N = 3010;

int dp[N][N][2];
int w[N][20];
int p[N];

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m;
    cin >> n >> m;
    int sum = 0;
    for(int i = 1; i <= n; i ++) {
        cin >> p[i];
        sum += p[i];
        for(int j = 1; j <= p[i]; j ++) cin >> w[i][j];
    }
    if(sum <= m) {
        int ans = 0;
        for(int i = 1; i <= n; i ++) ans += w[i][p[i]];
        cout << ans << "\n";
        return 0;
    }
    memset(dp, -0x3f, sizeof dp);
    dp[0][0][0] = 0;
    for(int i = 1; i <= n; i ++) {
        for(int j = 0; j <= m; j ++) {
            dp[i][j][0] = dp[i - 1][j][0];
            dp[i][j][1] = dp[i - 1][j][1];
            if(j >= p[i]) {
                dp[i][j][0] = max(dp[i][j][0], dp[i - 1][j - p[i]][0] + w[i][p[i]]);
                dp[i][j][1] = max(dp[i][j][1], dp[i - 1][j - p[i]][1] + w[i][p[i]]);
            }
            for(int k = 1; k < p[i]; k ++) {
                if(j >= k) dp[i][j][1] = max(dp[i][j][1], dp[i - 1][j - k][0] + w[i][k]);
            }
        }
    }
    cout << max(dp[n][m][0], dp[n][m][1]) << "\n";
}
相关推荐
琼羽10919 分钟前
第十七周-通用量子门与Deutsch-Jozsa算法
算法·量子计算
旺小仔.20 分钟前
位运算专题
算法
xwl12121 小时前
10.6 作业
数据结构·算法
青草地溪水旁1 小时前
VSCode C/C++ 构建任务配置文件 `tasks.json` 全字段深度解析
c语言·c++·vscode
爱和冰阔落3 小时前
C++模板进阶 非类型模板参数 模板的特化 分离编译的深入探索
c++·面试·编译原理·模板
胡小禾3 小时前
JDK17和JDK8的 G1
jvm·算法
胖咕噜的稞达鸭5 小时前
算法入门:专题攻克一---双指针(3)有效三角形的个数 查找总价格为目标值的两个商品(剑指offer题目)
算法
charlie1145141919 小时前
精读C++20设计模式:行为型设计模式:中介者模式
c++·学习·设计模式·c++20·中介者模式
楼田莉子9 小时前
Qt开发学习——QtCreator深度介绍/程序运行/开发规范/对象树
开发语言·前端·c++·qt·学习
逻辑留白陈9 小时前
Adaboost进阶:与主流集成算法对比+工业级案例+未来方向
算法