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-8using 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;}