C++动态求连续区间和(线段树)

给定 n 个数组成的一个数列,规定有两种操作,一是修改某个元素,二是求子数列 [a,b]的连续和。
输入格式
第一行包含两个整数 n和 m,分别表示数的个数和操作次数。
第二行包含 n个整数,表示完整数列。
接下来 m行,每行包含三个整数 k,a,b (k=0,表示求子数列[a,b]和;k=1,表示第 a 个数加 b)。
数列从 1开始计数。
输出格式
输出若干行数字,表示 k=0时,对应的子数列 [a,b]的连续和。
数据范围
1≤n≤100000,1≤m≤100000,1≤a≤b≤n
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8
输出样例:
11
30
35

AC代码:

#include<stdio.h>

const int N=100010;
//线段树最多有4n个结点
int a[N],tree[4*N];
//用a[l]到a[r]为线段树构建节点,父节点是node
void build(int node,int l,int r)
{
    if(l==r){//l,r是原数组下标,node是线段树节点号
        tree[node]=a[l];
    }
    else {
        int mid=l+((r-l)>>1);
        int left_node=node<<1;
        int right_node=(node<<1)|1;
        build(left_node,l,mid);
        build(right_node,mid+1,r);
        tree[node]=tree[left_node]+tree[right_node];
    }
}
//给a[idx]加上val
void add(int node,int l,int r,int idx,int val)
{
    if(l==r){
        tree[node]+=val;
        //a[idx]+=val;
        //如果加上这句,还可以实现O(1)查询某个元素
    }
    else {
        int mid=l+((r-l)>>1);
        int left_node=node<<1;
        int right_node=(node<<1)|1;
        //根据要修改元素下标在哪个区间范围判断要修改哪边
        if(idx>=l&&idx<=mid) add(left_node,l,mid,idx,val);
        else add(right_node,mid+1,r,idx,val);
        //维护父节点
        tree[node]=tree[left_node]+tree[right_node];
    }
}
//a,b是所求连续区间左右端点,l,r是当前查询区间
int query(int node,int l,int r,int a,int b)
{
    //当前查询区间不在所求范围内,不用计算
    if(a>r||b<l) return 0;
    //当前区间在所求区间内,或搜索到单个元素,当前节点值可用
    else if(a<=l&&b>=r||l==r) return tree[node];
    else {
        int mid=l+((r-l)>>1);
        int left_node=node<<1;
        int right_node=(node<<1)|1;
        int left_sum=query(left_node,l,mid,a,b);
        int right_sum=query(right_node,mid+1,r,a,b);
        return left_sum+right_sum;
    }
}

int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i) scanf("%d",&a[i]);
    //根节点,初始区间范围
    build(1,1,n);
    int k,x,y;
    while(m--)
    {
        scanf("%d%d%d",&k,&x,&y);
        if(k==0) printf("%d\n",query(1,1,n,x,y));
        else add(1,1,n,x,y);
    }
    return 0;
}
//本模板记忆窍门:
//node l r必有,剩下两个参数看情况
//mid left_node right_node必求
  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

赴星辰大海

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值