线段树

来源:互联网 发布:淘宝怎样获得天猫积分 编辑:程序博客网 时间:2024/06/06 00:52
#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1long long sum[MAX<<2],col[MAX<<2];void pushup(int rt)//本身不是动态的过程,在递归过程中向上更新{    sum[rt]=sum[rt<<1]+sum[rt<<1|1];}void pushdown(int rt,int m){    if(col[rt])    {        col[rt<<1]+=col[rt];        col[rt<<1|1]+=col[rt];        sum[rt<<1]+=(m-(m>>1))*col[rt];        sum[rt<<1|1]+=(m>>1)*col[rt];        col[rt]=0;    }}void build(int l,int r,int rt)//建树build(1,n,1){    col[rt]=0;    if(l==r)    {        scanf("%lld",&sum[rt]);//灵活        return;    }    int m=(l+r)>>1;    build(lson);    build(rson);    pushup(rt);//向上更新}long long query(int L,int R,int l,int r,int rt)//查询操作{    if(L<=l&&r<=R)        return sum[rt];    pushdown(rt,r-l+1);    int m=(l+r)>>1;    long long ret=0;    if(L<=m)//有当前区间的左区间部分        ret+=query(L,R,lson);    if(R>m)        ret+=query(L,R,rson);    return ret;}void update(int L,int R,int c,int l,int r,int rt){        if(L<=l&&r<=R)        {            col[rt]+=c;            sum[rt]+=c*(r-l+1);            return;        }        pushdown(rt,r-l+1);//延时更新        int m=(l+r)>>1;        if(L<=m)            update(L,R,c,lson);        if(R>m)            update(L,R,c,rson);        pushup(rt);}
0 0
原创粉丝点击