CodeForces 75D Big Maximum Sum （线段树辅助DP）

题目链接：http://codeforces.com/problemset/problem/75/D

#include<bits/stdc++.h>
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define ll  __int64

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
#define mst(a,b) memset((a),(b),sizeof(a))

#define pii pair<ll,ll>
#define mk(x,y) make_pair(x,y)

const int maxn =50+5;
const int maxm=250005;
const int mod=1e9+7;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
题目大意：给定若干个序列，和一串编号序列，
然后求整体序列的最大子段和。

首先可以讨论出答案的存在有两种情况，
一种是一端有前缀，往后是完全覆盖，
一种是两端有前缀，中间是完全覆盖，
那么对于每个序列，我们要提取出来的东西有，前缀最大和，后缀最大和，
最大子段和，和整体和。
第一种情况很好筛选答案，只要前缀和形式累加下就行。
第二种情况，需要线段树维护下区间修改，

时间复杂度：O(mlogm).


*/

ll n,m,k,a[maxm],q[maxm],tmp;
ll l[maxn],r[maxn],mid[maxn],sum[maxn];
void getm(int len,int idx){
    ll tot=0,ans=sum[idx];
    rep(i,1,len+1){
        tot+=a[i];
        ans=max(ans,tot);///这边顺序不能错。。。
        if(tot<=0)  tot=0;
    }
    mid[idx]=ans;
}
///线段树
ll tree[maxm<<2],lazy[maxm<<2];
void pushup(lrt){
    tree[rt]=max(tree[rt<<1],tree[rt<<1|1]);
}
void pushdown(lrt){
    if(lazy[rt]){
        tree[rt<<1]+=lazy[rt];
        tree[rt<<1|1]+=lazy[rt];
        lazy[rt<<1]+=lazy[rt];
        lazy[rt<<1|1]+=lazy[rt];
        lazy[rt]=0;
    }
}
void update(lrt,int L,int R,ll v){
    if(L>R) return ;
    if(L<=l&&r<=R){
        tree[rt]+=v;
        lazy[rt]+=v;
        return;
    }
    pushdown(root);
    int mid=l+r>>1;
    if(L<=mid) update(lson,L,R,v);
    if(mid<R) update(rson,L,R,v);
    pushup(root);
}
ll query(lrt,int L,int R){
    if(L>R) return 0LL;
    if(L<=l&&r<=R)return tree[rt];
    pushdown(root);
    int mid=l+r>>1;
    ll ans=0;
    if(L<=mid) ans=max(ans,query(lson,L,R));
    if(mid<R) ans=max(ans,query(rson,L,R));
    return ans;
}
int main(){
    cin>>n>>m;
    ///scanf("%I64d%I64d",&n,&m);///输入序列个数和总序列编号
    rep(i,1,n+1){
        ll lmax=0,rmax=0,mmax=0,tot=0;
        cin>>k;
        rep(j,1,k+1){
            cin>>a[j];///输入序列
            tot+=a[j];
        }
        lmax=rmax=mmax=tot;
        sum[i]=tot;

        tmp=0;///暂时变量
        rep(j,1,k+1){
            tmp+=a[j];
            lmax=max(tmp,lmax);
        }
        tmp=0;
        for(int j=k;j>=1;j--){
            tmp+=a[j];
            rmax=max(tmp,rmax);
        }
        l[i]=lmax,r[i]=rmax;///设置左右前缀数组
        getm(k,i);///设置最大子段和
    }

    mst(lazy,0),mst(tree,0);
    rep(i,1,m+1)  cin>>q[i];
    ll ans=mid[q[1]];tmp=0;
    rep(i,1,m+1){
        ans=max(ans,mid[q[i]]);
        ans=max(ans,tmp+l[q[i]]);
        tmp+=sum[q[i]];
    }
    tmp=0;
    for(int i=m;i>=1;i--){
        ans=max(ans,tmp+r[q[i]]);
        tmp+=sum[q[i]];
    }
    ///下面筛选当前为前端的最大值。
    rep(i,1,m+1){
        ans=max(ans,query(1,maxm,1,1,i-1)+l[q[i]]);
        update(1,maxm,1,1,i-1,sum[q[i]]);
        update(1,maxm,1,i,i,r[q[i]]);
    }
    cout<<ans<<endl;
    return 0;
}