POJ 4047 Garden 线段树 区间更新

给出一个n个元素的序列，序列有正数也有负数

支持3个操作：

p x y

0.p=0时，把第x个的值改为y

1.p=1时，交换第x个和第y个的值

2.p=2时，问区间[x,y]里面连续k个的子序列的最大和(保证y-x+1>=k)

我们只要定义数组v

v[i]表示原序列中，从第i个开始，连续k个元素的值的和

然后我们只需要维护一棵线段树，树的叶子节点表示数组v

树的节点维护：

区间[l,r]中，连续k个的子序列的最大和，即数组v的最大值

这样的话，3个操作就变为：

0.把区间[max(x-k+1,0),x]的值加y-init_v[x]

1.区间[max(x-k+1,0),x]加上init_v[y]-init_v[x]

　区间[max(y-k+1,0),y]加上init_v[x]-init_v[y]

　交换init_v[x]和init_v[y]的值

2.求max[x,y-k+1]

 #include<cstdio>
 #include<cstring>
 #include<algorithm>

 using namespace std;

 #define ll long long
 #define lson l,m,rt<<1
 #define rson m+1,r,rt<<1|1

 const int maxn=+;
 const int inf=0x3f3f3f3f;

 int init_v[maxn];
 int init_sum[maxn];
 int v[maxn<<];
 int lazy[maxn<<];
 int n,m,k;

 void solve();

 int main()
 {
     int test;
     scanf("%d",&test);
     while(test--){
         scanf("%d %d %d",&n,&m,&k);
         for(int i=;i<=n;i++){
             scanf("%d",&init_v[i]);
         }
         solve();
     }
     return ;
 }

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

 void pushdown(int rt)
 {
     if(lazy[rt]){
         lazy[rt<<]+=lazy[rt];
         lazy[rt<<|]+=lazy[rt];
         v[rt<<]+=lazy[rt];
         v[rt<<|]+=lazy[rt];
         lazy[rt]=;
     }
 }

 void build(int l,int r,int rt)
 {
     if(l==r){
         v[rt]=init_sum[l+k-]-init_sum[l-];
         return ;
     }
     int m=(l+r)>>;
     build(lson);
     build(rson);
     pushup(rt);
 }

 void update(int L,int R,int add,int l,int r,int rt)
 {
     if(L<=l&&R>=r){
         lazy[rt]+=add;
         v[rt]+=add;
         return ;
     }
     int m=(l+r)>>;
     pushdown(rt);
     if(L<=m)
         update(L,R,add,lson);
     if(R>m)
         update(L,R,add,rson);
     pushup(rt);
 }

 int query(int L,int R,int l,int r,int rt)
 {
     if(L<=l&&R>=r){
         return v[rt];
     }
     int m=(l+r)>>;
     pushdown(rt);
     int ret=-inf;
     if(L<=m)
         ret=max(ret,query(L,R,lson));
     if(R>m)
         ret=max(ret,query(L,R,rson));

     return ret;
 }

 void solve()
 {
     init_sum[]=;
     for(int i=;i<=n;i++){
         init_sum[i]=init_sum[i-]+init_v[i];
     }

     build(,n,);
     memset(lazy,,sizeof lazy);
     for(int i=;i<=m;i++){
         int p,x,y;
         scanf("%d %d %d",&p,&x,&y);
         if(p==){
             update(max(x-k+,),x,y-init_v[x],,n,);
             init_v[x]=y;
         }
         else if(p==){
             update(max(x-k+,),x,init_v[y]-init_v[x],,n,);
             update(max(y-k+,),y,init_v[x]-init_v[y],,n,);
             swap(init_v[x],init_v[y]);
         }
         else{
             printf("%d\n",query(x,y-k+,,n,));
         }
     }
     return ;
 }