解决最大连续和问题可用dp[i]表示以a[i]结尾的最大连续和:
dp[0]=0
dp[i]=max(dp[i-1]+a[i],a[i])。
这题也也已用同样的原理,最大和来自两个部分,一个是各段内部的最大连续和,可用上述方法解决。对于拼起来的部分,令dp[i]表示以第i个数组结束的最大和,对每个数组预处理出,最大前缀和L[i],最大后缀和R[i],数组和sum[i],设大数组中的i个数组是原编号b[i]的数组,那么:
dp[i]=max(dp[i-1]+sum[b[i]],R[b[i]]),这样转移保证了以第i个数组的最后一个数结尾。
注意还要用dp[i-1]+L[b[i]]更新一下答案。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define maxn 250009
#define INF 1LL<<61
using namespace std;
int n,m;
int a[55][5009],len[55],b[maxn];
__int64 L[55],R[55],sum[55],pre[maxn],dp[maxn];
bool vis[55];
__int64 solve(int id)
{
__int64 res=-INF;
for(int i=1;i<=len[id];i++)
{
dp[i]=max(dp[i-1]+a[id][i],1LL*a[id][i]);
res=max(res,dp[i]);
}
return res;
}
int main()
{
__int64 ans=-INF;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&len[i]);
for(int j=1;j<=len[i];j++)
scanf("%d",&a[i][j]);
}
for(int i=1;i<=n;i++)
{
sum[i]=0;
L[i]=-INF;
for(int j=1;j<=len[i];j++)
{
sum[i]+=a[i][j];
L[i]=max(L[i],sum[i]);
}
}
for(int i=1;i<=n;i++)
{
sum[i]=0;
R[i]=-INF;
for(int j=len[i];j>=1;j--)
{
sum[i]+=a[i][j];
R[i]=max(R[i],sum[i]);
}
}
for(int i=1;i<=m;i++)
{
scanf("%d",&b[i]);
pre[i]=pre[i-1]+sum[b[i]];
vis[b[i]]=1;
}
for(int i=1;i<=m;i++)
{
ans=max(ans,max(L[b[i]],L[b[i]]+dp[i-1]));
dp[i]=max(R[b[i]],dp[i-1]+sum[b[i]]);
ans=max(ans,dp[i]);
}
for(int i=1;i<=n;i++)
{
if(vis[i])
ans=max(ans,solve(i));
}
printf("%I64d\n",ans);
//system("pause");
return 0;
}