敌兵布阵 模板

这里不想说什么,只是个博客么,不想说太多与线段树的东西,简单说说。

先建树,之后update,query,都是常规操作。

第二次写,从原来的108行减到62行,还是很高兴的。

#include <bits/stdc++.h>
#define ls rt<<1 
#define rs rt<<1|1
using namespace std ;
const int N = 50010 ;
struct node{int l,r,val,lazy;}tr[N<<2];
int a[N],T,n ;
char op[10] ;
void pushup(int rt){
    tr[rt].val=tr[ls].val+tr[rs].val ;
}
void build(int l,int r,int rt){
    tr[rt].l=l,tr[rt].r=r ;
    if (l==r){
        tr[rt].val=a[l] ;
        return ;
    } 
    int mid=(l+r)>>1 ;
    build(l,mid,ls) ;
    build(mid+1,r,rs) ;
    pushup(rt);
}
void update(int rt,int pos,int c){
    if (tr[rt].l==tr[rt].r){
        tr[rt].val+=c ;
        return ;
    }
    int mid=(tr[rt].l+tr[rt].r)>>1 ;
    if (pos<=mid) update(ls,pos,c) ;
    else update(rs,pos,c) ;
    pushup(rt) ;
}
int query(int rt,int l,int r){
    if (l==tr[rt].l && r==tr[rt].r) return tr[rt].val ;
    int mid=(tr[rt].l+tr[rt].r)>>1 ;
    int res=0 ;
    if (l<=mid && r>mid) res=query(ls,l,mid)+query(rs,mid+1,r) ;
    else if (l<=mid) res=query(ls,l,r) ; 
    else if (r>mid) res=query(rs,l,r) ;
    return res ;
}
int main(){
    scanf("%d",&T) ;
    for (int rnd=1;rnd<=T;rnd++){
        scanf("%d",&n) ;
        for (int i=1;i<=n;i++) scanf("%d",&a[i]) ;
        build(1,n,1) ;
        printf("Case %d:\n",rnd) ;
        while(scanf("%s",op)){
            if (op[0]=='E') break ;
            else {
                int x,y ;
                scanf("%d%d",&x,&y) ;
                if (op[0]=='A') update(1,x,y) ;
                else if (op[0]=='S') update(1,x,-y) ;
                else printf("%d\n",query(1,x,y)) ;      
            }
        }
    }
}

猜你喜欢

转载自blog.csdn.net/HQG_AC/article/details/81482657