poj3468(A simple Problem with Integers)(线段树成段增减)

利用懒惰标记使更新延迟到下一次需要更新or询问时。

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <algorithm>
#include <stdlib.h>
using namespace std;
#define ms(X) memset(X,0,sizeof(X));
typedef long long ll;
ll tree[120005<<2],cnt[120005<<2];/*感觉cnt是int也可以啊,因为最大是10000*100000,不会爆int啊,然而wa了......*/
int N,Q;
void build(int tn,int l,int r)
{
    if(l==r) {scanf("%lld",tree+tn);return;}
    int mid=(l+r)>>1;
    build(tn<<1,l,mid);
    build(tn<<1|1,mid+1,r);
    tree[tn]=tree[tn<<1]+tree[tn<<1|1];
}
void pushdown(int tn,int m)
{
    if(cnt[tn]&&m>1)
    {
        cnt[tn<<1]+=cnt[tn];
        cnt[tn<<1|1]+=cnt[tn];
        tree[tn<<1]+=(m-(m>>1))*cnt[tn];
        tree[tn<<1|1]+=(m>>1)*cnt[tn];
        cnt[tn]=0;
    }
}
void chang(int tn,int l,int r,int a,int b,int c)
{
    if(a<=l&&r<=b){
        cnt[tn]+=c;
        tree[tn]+=(ll)c*(r-l+1);
        return;
    }
    pushdown(tn,r-l+1);
    int mid=(l+r)>>1;
    if(b<=mid) chang(tn<<1,l,mid,a,b,c);
    else if(a>mid) chang(tn<<1|1,mid+1,r,a,b,c);
    else{
        chang(tn<<1,l,mid,a,mid,c);
        chang(tn<<1|1,mid+1,r,mid+1,b,c);
    }
    tree[tn]=tree[tn<<1]+tree[tn<<1|1];
}
ll query(int tn,int l,int r,int a,int b)
{
    if(a<=l&&r<=b) return tree[tn];
    pushdown(tn,r-l+1);
    int mid=(l+r)>>1;
    if(b<=mid) return query(tn<<1,l,mid,a,b);
    else if(a>mid) return query(tn<<1|1,mid+1,r,a,b);
    else return query(tn<<1,l,mid,a,mid)+query(tn<<1|1,mid+1,r,mid+1,b);
}
int main(int argc, char const *argv[])
{
    while(cin>>N>>Q)
    {
        ms(cnt);
        build(1,1,N);
        char ss[3];
        while(Q--){scanf("%s",ss);
        if(ss[0]=='C'){
            int a,b,c;
            scanf("%d %d %d",&a,&b,&c);
            chang(1,1,N,a,b,c);
        }
        else {
            int a,b;
            scanf("%d %d",&a,&b);
            printf("%lld\n",query(1,1,N,a,b));
        }}
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值