1264. 动态求连续区间和 【模板】【树状数组】

1264. 动态求连续区间和

给定 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

树状数组

空间复杂度:O(n),和原始的数组大小一样
时间复杂度:在某一点加值logN,查询某个区间logN
1.每个结点tr[x] 保存的是区间 (x-lowbit(x),x] 的和
2.每个结点tr[x]的父结点是tr[x+lowbit(x)],当一个结点所管理的区间发生改变,就会影响到其父结点,复杂度logN
3.修改操作只能是在某个点+一个值,而不能是直接改变,所要改变可以换成加差值,tr[x] += v-c(v要变成的值,c原来的值)
4.树状数组可以解决的问题都可以用线段树解决,这两者的区别在哪里呢?树状数组的系数要少很多,就比如字符串模拟大数可以解决大数问题,也可以解决1+1的问题,但没人会在1+1的问题上用大数模拟

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e6+10;

int N,M;
int tr[maxn];

int lowbit(int x){
    return x&-x; //返回只保留二进制的最后一个1之后的值
}
void add(int idx,int x){ //向下标为idx的元素+x,同时更新其所影响结点的前缀和
    for(int i = idx;i<=N;i += lowbit(i)) //这里的N是数组最后一个下标,有时候不一定是N
        tr[i] += x;
}

int query(int idx){ //求下标为idx的前缀和
    int sum = 0;
    for(int i = idx;i>=1;i -= lowbit(i))
        sum += tr[i];
    return sum;
}

int main(){
    cin>>N>>M;
    int tmp;
    for(int i = 1;i<=N;i++){
        scanf("%d",&tmp);
        add(i,tmp);
    }
    int op,x,y;
    while(M--){
        scanf("%d%d%d",&op,&x,&y);
        if(op == 1) add(x,y);
        else printf("%d\n",query(y)-query(x-1));//前缀和的差 = 区间的和
    }

    return 0;
}
全部评论

相关推荐

牛客263158796号:我领羊一面后十天不挂也不推进 今天问hr说等前序的第一批意向发完看情况再看是否推进
点赞 评论 收藏
分享
斑驳不同:还为啥暴躁 假的不骂你骂谁啊
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务