【POJ】4047 Garden

#include<cstdio>
#define MAXN 200010
#define INF 987654321
int a[MAXN],s[MAXN],tree[MAXN<<2],lazy[MAXN<<2],k;
inline int MAX(int x,int y)
{
    return x>y?x:y;
}
inline int MIN(int x,int y)
{
    return x>y?y:x;
}
inline void swap(int &x,int &y)
{
    int temp=x;
    x=y;
    y=temp;
}
inline void PushUp(int rt)
{
    tree[rt]=MAX(tree[rt<<1],tree[rt<<1|1]);
}
void Build(int L,int R,int rt)
{
    lazy[rt]=0;
    if(L==R)
        tree[rt]=s[L+k-1]-s[L-1];
    else
    {
        int mid=(L+R)>>1;
        Build(L,mid,rt<<1);
        Build(mid+1,R,rt<<1|1);
        PushUp(rt);
    }
}
inline void PushDown(int rt)
{
    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(int x,int y,int z,int L,int R,int rt)
{
    if(x<=L&&R<=y)
    {
        lazy[rt]+=z;
        tree[rt]+=z;
    }
    else
    {
        int mid=(L+R)>>1;
        PushDown(rt);
        if(mid>=x)
            Update(x,y,z,L,mid,rt<<1);
        if(y>mid)
            Update(x,y,z,mid+1,R,rt<<1|1);
        PushUp(rt);
    }
}
int Query(int x,int y,int L,int R,int rt)
{
    if(x<=L&&R<=y)
        return tree[rt];
    int ans=-INF,mid=(L+R)>>1;
    PushDown(rt);
    if(mid>=x)
        ans=MAX(ans,Query(x,y,L,mid,rt<<1));
    if(y>mid)
        ans=MAX(ans,Query(x,y,mid+1,R,rt<<1|1));
    return ans;
}
int main()
{
    int n,m,c,i,p,x,y;
    scanf("%d",&c);
    while(c--)
    {
        scanf("%d%d%d",&n,&m,&k);
        s[0]=0;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            s[i]=s[i-1]+a[i];
        }
        n-=k-1;
        Build(1,n,1);
        while(m--)
        {
            scanf("%d%d%d",&p,&x,&y);
            if(p==0)
            {
                if(a[x]!=y)
                {
                    Update(MAX(1,x-k+1),x,y-a[x],1,n,1);
                    a[x]=y;
                }
            }
            else if(p==1)
            {
                if(x!=y)
                {
                    Update(MAX(1,x-k+1),x,a[y]-a[x],1,n,1);
                    Update(MAX(1,y-k+1),y,a[x]-a[y],1,n,1);
                    swap(a[x],a[y]);
                }
            }
            else
                printf("%d\n",Query(x,y-k+1,1,n,1));
        }
    }
    return 0;
}