线段树模板

#include <stdio.h>
#include <math.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <algorithm>
#include <iostream>
#include <ctype.h>
#include <set>
#include <map>
#include <vector>
#include <stack>
#include <queue>
#define left (now<<1)
#define right ((now<<1)+1)
#define mid ((l+r)>>1)
#define fst first
#define snd second
using namespace std;
typedef long long lint;


struct segmentTree{
    int sum;
};


int t,n,a[50010];
char str[10];
segmentTree tree[2000010];


void buildTree(int now,int l,int r){   //建树
    if(l == r){
        tree[now].sum = a[l]; return;
    }
    buildTree(left,l,mid); buildTree(right,mid+1,r);
    tree[now].sum = tree[left].sum + tree[right].sum;
    return;
}


void updateTree(int now,int l,int r,int set,int setNum){  //更新树
    if(l == r){
        tree[now].sum += setNum;
        return;
    }
    if(set <= mid){
        updateTree(left,l,mid,set,setNum);
    }else{
        updateTree(right,mid+1,r,set,setNum);
    }
    tree[now].sum = tree[left].sum + tree[right].sum;
}


int getSum(int now,int l,int r,int ql,int qr){   //求和
   if(ql == l && qr == r){
       return tree[now].sum;
   } 
   int re = 0;
   if(ql <= mid){
       re += getSum(left,l,mid,ql,min(qr,mid));
   }
   
   if(qr > mid){
       re += getSum(right,mid+1,r,max(ql,mid+1),qr);
   }
   return re;
}


int main(){
    scanf("%d",&t);
    int ca = 0;
    while(t--){
        ++ca;
        memset(a,0,sizeof(a));
        scanf("%d",&n);
        for(int i = 1; i <= n; ++i){
            scanf("%d",&a[i]);
        }
        scanf("%s",str);
        buildTree(1,1,n);
        printf("Case %d:n",ca);
        while(str[0] != 'E'){
            if(str[0] == 'A'){
                int u,v;
                scanf("%d%d",&u,&v);
                updateTree(1,1,n,u,v);
            }else if(str[0] == 'S'){
                int u,v;
                scanf("%d%d",&u,&v);
                updateTree(1,1,n,u,-v);
            }else if(str[0] == 'Q'){
                int u,v;
                scanf("%d%d",&u,&v);
                int ans = getSum(1,1,n,u,v);
                printf("%dn",ans);
            }
            scanf("%s",str);
        }
    }
}


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值