POJ 4047: Garden

线段树模板题

哎，好久没写线段树了，吃力。

题目链接：http://poj.org/ProblemDescriptions/jinghua.pdf

题意：

对某个区间进行以下三种操作：

0、改变某个位置的值

1、交换某两个位置的值

2、查询某个区间内连续k个数和最大值

算法：

由于k的值是确定的，

所以直接用每个叶子节点保存它之前的k个元素的值即可。

#include<cstdio>
#include<iostream>
#include<sstream>
#include<cstdlib>
#include<cstring>
#include<string>
#include<climits>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define eps 1e-8
using namespace std;

const int MAXN=210000;
int a[MAXN],sum[MAXN];
int tr[MAXN<<2],ret[MAXN<<2];

void pushup(int rt) {
    tr[rt]=max(tr[rt<<1],tr[rt<<1|1]);
}

void pushdown(int rt) {
    if(ret[rt]) {
        tr[rt<<1]+=ret[rt];
        tr[rt<<1|1]+=ret[rt];
        ret[rt<<1]+=ret[rt];
        ret[rt<<1|1]+=ret[rt];
        ret[rt]=0;
    }
}

void build(int l, int r, int rt) {
    ret[rt]=0;
    if(l==r) {
        tr[rt]=sum[l];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}

void update(int l, int r, int L, int R, int c, int rt) {
    if(L<=l&&r<=R) {
        tr[rt]+=c;
        ret[rt]+=c;
        return;
    }
    pushdown(rt);
    int mid=(l+r)>>1;
    if(L<=mid) {
        update(l,mid,L,R,c,rt<<1);
    }
    if(R>mid) {
        update(mid+1,r,L,R,c,rt<<1|1);
    }
    pushup(rt);
}

int query(int l, int r, int L, int R, int rt) {
    if(L<=l&&r<=R) {
        return tr[rt];
    }
    pushdown(rt);
    int mid=(l+r)>>1;
    int tmp=INT_MIN;
    if(L<=mid) {
        tmp=max(tmp,query(l,mid,L,R,rt<<1));
    }
    if(R>mid) {
        tmp=max(tmp,query(mid+1,r,L,R,rt<<1|1));
    }
    return tmp;
}

int main() {
    int cas;
    scanf("%d",&cas);
    while(cas--) {
        int n,m,k;
        scanf("%d%d%d",&n,&m,&k);
        for(int i=0; i<n; i++) {
            scanf("%d",&a[i]);
            sum[i]=a[i];
            if(i) {
                sum[i]+=sum[i-1];
            }
            if(i>=k) {
                sum[i]-=a[i-k];
            }
        }
        build(0,n-1,1);
        while(m--) {
            int p,x,y;
            scanf("%d%d%d",&p,&x,&y);
            if(p==0) {
                x--;
                update(0,n-1,x,x+k-1,y-a[x],1);
                a[x]=y;
            } else if(p==1) {
                x--;
                y--;
                int tmp=a[x];
                update(0,n-1,x,x+k-1,a[y]-a[x],1);
                a[x]=a[y];
                update(0,n-1,y,y+k-1,tmp-a[y],1);
                a[y]=tmp;
            } else {
                x--;
                y--;
                printf("%d\n",query(0,n-1,x+k-1,y,1));
            }
        }
    }
    return 0;
}