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;
}