牛客 F - 爬塔 （分组背包）

中国计量大学现代科技学院第四届“中竞杯”程序设计校赛（同步赛）F - 爬塔
链接：https://ac.nowcoder.com/acm/contest/9680/F

题目描述

高川最喜欢的游戏当属 Slay the Spire，这是一款爬塔游戏，你需要从一座塔的底部一直爬到顶部，在爬塔的过程中，塔的每一层都有许多的宝物等你来拿。

高川从塔的左侧开始攀爬，从底部爬到顶部，再从右侧从顶部逐步下到底部。塔总共有 n 层，每一层都有很多宝物从左到右排列。在左侧攀爬时，他只能从每层的最左边按顺序取宝物，在右侧下降时，他只能从每层的最右边按顺序取宝物。每个宝物都有一个价值，他最多拿 m 个宝物，他想知道自己从塔上下来时，最多可以拿的宝物价值和是多少。

输入描述:

第一行输入两个正整数 n 和 m。表示塔的层数和最多能选的个数。

接下来 n 行,每行先输入一个数字 x。表示这一层宝物的个数。接下来输入 x 个正整数，表示每个宝物的权重 c。

1≤n,x,c≤100
1≤m≤10000

输入保证可挑选的物品大于等于 m 个.

输出描述:

输出一个整数表示高川能拿走的最大价值和.

输入

2 3
2 3 2
4 1 4 1 5

输出

10

#include <bits/stdc++.h>

#define IO ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
using namespace std;
typedef long long ll;
typedef pair<long long, long long> pll;

int n, m;
int sz[110]; // 每一层上的宝物的个数
ll sum[110][110]; // 每一层的前缀和
vector<pll> bb[110]; // pair<个数v, 总和w>
ll f[15000]; // f[k][v]表示前k组物品花费v能取得的最大权值，压缩掉第一维

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &sz[i]);
        for (int j = 1; j <= sz[i]; ++j) {
            ll x;
            scanf("%lld", &x);
            sum[i][j] = sum[i][j - 1] + x;
        }
    }
    /* 因为分组背包的每一组中的两两物品都是互斥的，每组只能选一个
       所以我们把每一层作为一组，长度不同的各个物品总和就是互斥的物品 */
    for (int i = 1; i <= n; ++i) { // 第i层
        for (int j = 1; j <= sz[i]; ++j) { // 取j个
            int nj = sz[i] - j;  // 不取nj个
            if (nj == 0) {
                bb[i].emplace_back(j, sum[i][j]);
                continue;
            }
            ll tmp = 0;
            for (int k = 1; k + nj - 1 <= sz[i]; ++k) { // 枚举不取的 从k到k+nj-1
                tmp = max(tmp, sum[i][k - 1] + sum[i][sz[i]] - sum[i][k + nj - 1]);
            }
            bb[i].emplace_back(j, tmp);
        }
    }
    /*
    for 所有的组k
        for v=V..0
            for 所有的i属于组k
                f[v]=max{f[v],f[v-c[i]]+w[i]}
    */
    for (int i = 1; i <= n; ++i) { // n层相当于n组
        int siz = bb[i].size();
        for (int j = m; j >= 0; --j) { // m相当于体积，选几个宝物
            for (int k = 0; k < siz; ++k) { // 第i组的所有宝物
                if (j >= bb[i][k].first) {
                    f[j] = max(f[j], f[j - bb[i][k].first] + bb[i][k].second);
                }
            }
        }
    }
    printf("%lld\n", f[m]);
    return 0;
}