洛谷-3374 【模板】树状数组 1

最新推荐文章于 2024-04-02 16:05:49 发布

mkopvec

最新推荐文章于 2024-04-02 16:05:49 发布

阅读量128

点赞数

CC 4.0 BY-SA版权

分类专栏：算法题模板

本文链接：https://blog.youkuaiyun.com/mkopvec/article/details/90260930

算法题同时被 2 个专栏收录

311 篇文章

订阅专栏

模板

87 篇文章

订阅专栏

本文介绍了一种使用树状数组解决数列操作问题的方法，包括修改数列中的某个元素和查询某区间内所有元素的和。通过树状数组，可以高效地完成这两种操作，适用于大规模数据处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述
如题，已知一个数列，你需要进行下面两种操作：
1.将某一个数加上x
2.求出某区间每一个数的和
输入输出格式
输入格式：
第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。
第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。
接下来M行每行包含3个整数，表示一个操作，具体如下：
操作1：格式：1 x k 含义：将第x个数加上k
操作2：格式：2 x y 含义：输出区间[x,y]内每个数的和
输出格式：
输出包含若干行整数，即为所有操作2的结果。

输入输出样例
输入样例#1：
5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4

输出样例#1：
14
16

说明
时空限制：1000ms,128M
数据规模：
对于30%的数据：N<=8，M<=10
对于70%的数据：N<=10000，M<=10000
对于100%的数据：N<=500000，M<=500000

#include<cstdio>
#include<iostream>
#define lowbit(x) x&(-x)
#define N 500003
using namespace std;
int a[N]={0};
int temp=0;
void add(int pos,int val){
    for(int i=pos;i<N;i+=lowbit(i)){
        a[i]+=val;
    }
}
int sum(int pos){
    int ret=0;
    for(int i=pos;i>0;i-=lowbit(i)){
        ret+=a[i];
    }
    return ret;
}
int main(){
    int n=0,m=0;scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&temp);
        add(i,temp);
    }
    while(m--){
        int a=0,b=0,c=0;
        scanf("%d%d%d",&c,&a,&b);
        if(c==1){
            add(a,b);
        }else{
            printf("%d\n",sum(b)-sum(a-1));
        }
    }
    return 0;
}