线段树的建立、更新、查找

#include<iostream>
using namespace std;
struct node{
    int left,right;
    int sum;
}tree[100];
int p,v;
void Build(int l,int r,int cur){
    tree[cur].left=l;
    tree[cur].right=r;
    if(l==r){///到达单位区间
        cin>>tree[cur].sum;///直接给sum赋值
        return ;
    }
    int mid=(l+r)/2;
    Build(l,mid,cur*2);///建立左儿子节点
    Build(mid+1,r,cur*2+1);///建立右儿子节点
    tree[cur].sum=tree[cur*2].sum+tree[cur*2+1].sum;///递归回去的时候计算各节点sum值
}
void Update(int cur){///原理和Build()一样,所以有的时候不写Build(),写Update就足够了
    if(tree[cur].left==tree[cur].right&&tree[cur].left==p){///到达单位区间,并且是p所在的单位区间
        tree[cur].sum=v;///直接更改sum值
        return ;
    }
    int mid=(tree[cur].left+tree[cur].right)/2;
    if(p<=mid)Update(cur*2);///p在左儿子节点
    else Update(cur*2+1);///p在右儿子节点
    tree[cur].sum=tree[cur*2].sum+tree[cur*2+1].sum;///递归回去的时候更新相关节点的sum值
}
int Query(int l,int r,int cur){
    if(l==tree[cur].left&&r==tree[cur].right)///查询区间和节点的区间完全相同
        return tree[cur].sum;///直接返回该节点的sum值
    int mid=(tree[cur].left+tree[cur].right)/2;
    if(r<=mid)return Query(l,r,cur*2);///查询区间完全在左儿子节点的区间,则查询左儿子节点
    else if(l>mid)return Query(l,r,cur*2+1);///查询区间完全右儿子节点的区间,则查询右子节点
    else return Query(l,mid,cur*2)+Query(mid+1,r,cur*2+1);///查询区间在左右儿子节点区间都有,则都查
}
int main(){
    int n,l,r;
    cin>>n;
    Build(1,n,1);///建立区间[1,n],从根节点1开始
    cin>>p>>v;///将p点的值改成v
    Update(1);///从根节点1开始更新
    cin>>l>>r;
    cout<<Query(l,r,1);///查询区间[l,r],从根节点1开始
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值