poj 3468

题目:http://poj.org/problem?id=3468

成段更新线段树。用mark延迟标记,在更新的时候不用每次把叶子节点全部更新,只需要把需要更新的一段所需要更新的权值标记一下,然后在下次查询的时候。如果需要更新这个被标记的子节点。那么把这个子节点的儿子节点的延迟mark加上父亲节点的mark。


下面是AC代码:

#include<cstdio>
using namespace std;
#define maxn 100000+10
typedef long long LL;
struct node{
    int l,r,m;
    LL sum,mark;
}T[maxn<<2];
int a[maxn];
void build(int id,int l,int r){
     T[id].l=l;   T[id].r=r;  T[id].m=(l+r)>>1;  T[id].mark=0;
     if(l==r)   { T[id].sum=a[l]; return;  }
     int m=(l+r)>>1;
     build(id<<1,l,m);  build((id<<1)+1,m+1,r);
     T[id].sum=(T[id<<1].sum+T[(id<<1)+1].sum);
}
void update(int id,int l,int r,int val){
     if(T[id].l==l&&T[id].r==r){
        T[id].mark+=val; return ;
     }
     T[id].sum+=(LL)val*(r-l+1);
     if(T[id].m>=r)
          update(id<<1,l,r,val);
     else if(T[id].m<l)
          update((id<<1)+1,l,r,val);
     else{
          update(id<<1,l,T[id].m,val);
          update((id<<1)+1,T[id].m+1,r,val);
     }
}
LL query(int id,int l,int r){
    if(T[id].l==l&&T[id].r==r)  return T[id].sum+T[id].mark*(LL)(r-l+1);
    if(T[id].mark!=0) {
        T[id<<1].mark+=T[id].mark;
        T[(id<<1)+1].mark+=T[id].mark;
        T[id].sum+=(LL)(T[id].r-T[id].l+1)*T[id].mark;  T[id].mark=0;
    }

    if(T[id].m>=r){
          return query(id<<1,l,r);
    }
    else if(T[id].m<l){
          return query((id<<1)+1,l,r);
    }
    else{
          return query(id<<1,l,T[id].m)+query((id<<1)+1,T[id].m+1,r);
    }
}
int main(){
    int n,Q;  char str[8];   int b,c,d;
    while(scanf("%d%d",&n,&Q)!=EOF){
          for(int i=1;i<=n;i++){
             scanf("%d",&a[i]);
          }
          build(1,1,n);
          for(int i=0;i<Q;i++){
               scanf("%s",str);
               if(str[0]=='Q'){
                    scanf("%d%d",&b,&c);
                    printf("%lld\n",query(1,b,c));
               }
               else if(str[0]=='C'){
                    scanf("%d%d%d",&b,&c,&d);
                    update(1,b,c,d);
               }
          }
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值