BZOJ 4094 USACO 2013 Dec. Optimal Milking

 

线段树

每个节点保存4个值，both表示左右端点都取，neither表示左右端点都不取，left表示只取左端点，right表示只取右端点。

维护的特殊姿势：

$cur$的$both=max(ls.l+rs.r，ls.both+rs.r，ls.l+rs.both)$

$cur$的$neither=max(ls.nei+rs.nei，ls.r+rs.nei，ls.nei+rs.l)$

$cur$的$left=max(ls.l+rs.l，ls.l+rs.nei，ls.both+ls.nei)$

$cur$的$right=max(ls.r+rs.r，ls.nei+rs.r，ls.nei+rs.both)$

#include<cstdio>
#include<algorithm>
#define LL long long
#define ls (cur<<1)
#define rs (cur<<1|1)
#define mid ((l+r)>>1)
using namespace std;
const int maxn=800010;
LL n,m,ans;
struct tree{
    LL both,l,r,nei;
}a[maxn];
void read(LL &k){
    k=0; int f=1; char c=getchar();
    while (c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while ('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    k*=f;
}
void pushup(int cur,int l,int r){
    if (l==r) return;
    int t1=a[ls].both+a[rs].r,t2=a[ls].l+a[rs].both,t3=a[ls].l+a[rs].r, //both
        t4=a[ls].l+a[rs].l,t5=a[ls].l+a[rs].nei,t6=a[ls].both+a[rs].nei, //left
        t7=a[ls].r+a[rs].r,t8=a[ls].nei+a[rs].r,t9=a[ls].nei+a[rs].both, //right
        t10=a[ls].nei+a[rs].nei,t11=a[ls].nei+a[rs].l,t12=a[ls].r+a[rs].nei; //neither
    a[cur].both=max(max(t1,t2),t3);
    a[cur].l=max(max(t4,t5),t6);
    a[cur].r=max(max(t7,t8),t9);
    a[cur].nei=max(max(t10,t11),t12);
}
void build(int cur,int l,int r){
    a[cur].l=0; a[cur].r=0; a[cur].nei=0;
    if (l<r){
        build(ls,l,mid);
        build(rs,mid+1,r);
    }
    else read(a[cur].both);
    pushup(cur,l,r);
}
void update(int cur,int l,int r,int pos,int x){
    if (l==r) a[cur].both=x;
    else{
        if (pos<=mid) update(ls,l,mid,pos,x);
        else update(rs,mid+1,r,pos,x);
    }
    pushup(cur,l,r);
}
int main(){
    read(n); read(m);
    build(1,1,n);
    for (int i=1;i<=m;i++){
        LL x,y; read(x); read(y);
        update(1,1,n,x,y);
        ans+=max(max(max(a[1].both,a[1].nei),a[1].l),a[1].r);
    }
    printf("%lld\n",ans);
    return 0;
}