codeforces 1442 D Sum 可撤销01背包QwQ

原创已于 2022-04-02 12:16:35 修改 · 653 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#操作系统 #分布式

于 2020-11-09 20:44:16 首次发布

acm 同时被 2 个专栏收录

78 篇文章

订阅专栏

Codeforces

28 篇文章

订阅专栏

本文介绍了一种针对多个单调不减数组选取元素求最大和的问题，并给出了一种分治优化的方法，将复杂度降至O(nklog)，通过枚举不全选的数组并结合01背包动态规划技巧，实现高效求解。

link
题意：
有n个单调不减数组，求从这n个数组中选择k个数的最大和，（只能拿数组的一个前缀）
n,k<3000
思路：因为数组是单调的，所以能得到一个结论就是我们最多只会拿不全一个数组（当有两个数组都拿一部分时，那么肯定可以通过减少某个数组拿的数字去拿另一个数组来增大这个和）。
所以我们就可以通过枚举哪个数组不全选，然后对剩下的数组跑01背包，在枚举当前数组选几个来更新答案，但可惜这是O(nnk)的，
有一个巧妙的优化就是，我们分治整个区间，对于一个l,r维护不选这个区间内的数字的dp值。那么当到达l==r时，就可以直接更新了。O(nklog)
当到达分治的第K层l,r时，我们对k+1层的dp值进行l,mid的更新，然后去递归mid+1,r的区间，然后再回溯的时候在把第k+1层的dp值重新赋值成k层的dp，再对k+1层的dp进行mid+1,r的更新，递归l，mid。
具体实现看代码：

#include<bits/stdc++.h>
#define pb push_back
using namespace std;
typedef long long ll;
const int maxn = 3e3 + 5;


ll dp[20][maxn],c[maxn],a[maxn][maxn];
ll ans;
int n,k;
void init(int d) {
    for(int i=0;i<=k;i++) dp[d][i]=dp[d-1][i];
}
void upd(int d,int l,int r) {
    for(int i=l;i<=r;i++) {
        for(int j=k;j>=c[i];j--) {
            dp[d][j]=max(dp[d][j],dp[d][j-c[i]]+a[i][c[i]]);
        }
    }
}

void sol(int d,int l,int r) {
    if(l==r) {
        ans=max(ans,dp[d][k]);
        for(int i=1;i<=c[l]&&i<=k;i++) {
            ans=max(ans,dp[d][k-i]+a[l][i]);
        }
        return ;
    }
    int mid=l+r>>1;
    init(d+1);
    upd(d+1,l,mid);
    sol(d+1,mid+1,r);

    init(d+1);
    upd(d+1,mid+1,r);
    sol(d+1,l,mid);
}

int main() {
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) {
        scanf("%d",&c[i]);
        for(int j=1;j<=c[i]&&j<=k;j++) {
            scanf("%lld",&a[i][j]);
            a[i][j]+=a[i][j-1];
        }
        for(int j=k+1,d;j<=c[i];j++) scanf("%d",&d);
    }
    sol(1,1,n);
    printf("%lld\n",ans);
    return 0;
}