【模板】树状数组

题自洛谷

题目描述

如题，已知一个数列，你需要进行下面两种操作：

1.将某一个数加上x

2.求出某区间每一个数的和

输入输出格式

输入格式：

第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数，表示一个操作，具体如下：

操作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

故输出结果14、16

就是树状数组的模板呀

树状数组是单点修改区间查询，线段树是区间修改区间查询

所以还是那句话，树状数组能做的线段树都可以写

但是线段树能做的树状数组不一定可以

但是树状数组好写啊，线段树那么长的代码

而且add函数和query函数很容易一不小心就哪个地方出错了，重点是一出错还难找

所以树状数组还是一个好东西的说

代码

#include<cstdio>
using namespace std;
int c[1000005],n,m;
inline int read()
{
    int sum=0; char ch=getchar();
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch<='9'&&ch>='0') sum=sum*10+ch-48,ch=getchar();
    return sum;
} 
int lowbit(int x)
{
    return x&-x;
}//树状数组中的核心
void add(int k,int num)
{
    while(k<=n)
    {
        c[k]+=num;
        k+=lowbit(k);
    }
}
int ask(int k)
{
    int ans=0;
    while(k>0)
    {
        ans+=c[k];
        k-=lowbit(k);
    }
    return ans;
}
int main()
{
    int x,y,z;
    n=read(),m=read();
    for(int i=1;i<=n;i++)
    x=read(),add(i,x);
    for(int i=1;i<=m;i++)
    {
        x=read(),y=read(),z=read();
        if(x==1) add(y,z);
        else printf("%d",ask(z)-ask(y-1)); //注意是y-1 
    }
    return 0;
}