【线段树】敌兵布阵-------实例入门理解

更新节点,区间求和

#include
#include 
using namespace std;
int sum;
int ac[50010];

struct node{
    int l,r,w;//区间左,右,区间和
}t[140000];

/*建树思路:a、对于二分到的每一个结点,给它的左右端点确定范围。
        b、如果是叶子节点,存储要维护的信息。
        c、状态合并。*/
void build(int k,int l,int r){

//k=1从1开始 (上到下)
    t[k].l=l;
    t[k].r=r;
    if(l==r){//叶子节点
        t[k].w=ac[l];
    }
    else{
        int mm=(l+r)/2;
        build(k*2,l,mm);//左孩子
        build(k*2+1,mm+1,r);//右孩子
        t[k].w=t[k*2].w+t[k*2+1].w;//状态合并,此结点的w=两个孩子的w之和 (更新节点信息)
    }    
}


//区间查询,即查询一段区间的状态,在引例中为查询区间[x,y]的和

/*与二分查询法基本一致,如果当前枚举的点左右端点相等,即叶子节点,就是目标节点。如果不是,因为这是二分法,所以设查询位置为x,当前结点区间范围为了l,r,中点为         mid,则如果x<=mid,则递归它的左孩子,否则递归它的右孩子*/


void query(int x,int y,int k){
 //x--y待查询区间 ,t[k]l--r当前区间
    if(x<=t[k].l && y>=t[k].r){
        //待查询区间完全包含当前区间
        sum+=t[k].w;
        
    }
    else{
        int mm=(t[k].l+t[k].r)/2;
        //查询区间全在当前区间的左子区间
         if(x>=mm+1) query(x,y,k*2+1);
        //查询区间全在当前区间的右子区间
        else if(y<=mm) query(x,y,k*2);
        else{
        //当前区间完全包含待查询区间
            query(x,y,k*2);
            query(x,y,k*2+1);
        }
    }
}

//单点修改,即更改某一个点的状态。用引例中的例子,对第x个数加上y
void add(int x,int y,int k){
    t[k].w+=y;
    if(t[k].l==t[k].r) return;
    int m=(t[k].l+t[k].r)/2;
    if(x>m) add(x,y,k*2+1);
    if(x<=m) add(x,y,k*2);
}

//单点修改,引例 对第x个数减去y
void sub(int x,int y,int k){
    t[k].w-=y;
    if(t[k].l==t[k].r) return ;
    int m=(t[k].l+t[k].r)/2;
    if(x>m) sub(x,y,k*2+1);
    if(x<=m) sub(x,y,k*2);    
}
int main(){
    int num;
    scanf("%d",&num);
        char ss[15];
        int n,a,b,yy=0;
        while(num--){
            scanf("%d",&n);
            for(int i=1;i<=n;i++)
            {
                scanf("%d",&ac[i]);
            }
            build(1,1,n);
            printf("Case %d:\n",++yy);
            while(1)
        {
            
            scanf("%s",ss);
        
            
            if(ss[0]=='E')
            
             break;
            
            if(ss[0]=='Q')
            {
                scanf("%d%d",&a,&b);
                sum=0;
                query(a,b,1);
                printf("%d\n",sum);
            }
            if(ss[0]=='A')
            {
                scanf("%d%d",&a,&b);
                add(a,b,1);
            }
            if(ss[0]=='S')
            {
            scanf("%d%d",&a,&b);
            sub(a,b,1);
            }
        
        }
    }
    return 0;
}

 

 

推荐参考大牛地址:https://www.cnblogs.com/TheRoadToTheGold/p/6254255.html

简单理解:https://blog.csdn.net/zearot/article/details/52280189(有关二进制)

你可能感兴趣的:(线段树)