P3374 【模板】树状数组 1

# 【模板】树状数组 1

## 题目描述

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

- 将某一个数加上 $x$

- 求出某区间每一个数的和

## 输入格式

第一行包含两个正整数 $n,m$，分别表示该数列数字的个数和操作的总个数。   
 
第二行包含 $n$ 个用空格分隔的整数，其中第 $i$ 个数字表示数列第 $i$ 项的初始值。

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

- `1 x k`  含义：将第 $x$ 个数加上 $k$

- `2 x y`  含义：输出区间 $[x,y]$ 内每个数的和

## 输出格式

输出包含若干行整数，即为所有操作 $2$ 的结果。

## 样例 #1

### 样例输入 #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
```

## 提示

【数据范围】

对于 $30\%$ 的数据，$1 \le n \le 8$，$1\le m \le 10$；   
对于 $70\%$ 的数据，$1\le n,m \le 10^4$；   
对于 $100\%$ 的数据，$1\le n,m \le 5\times 10^5$。

数据保证对于任意时刻，$a$ 的任意子区间（包括长度为 $1$ 和 $n$ 的子区间）和均在 $[-2^{31}, 2^{31})$ 范围内。

样例说明：

  

故输出结果14、16

 

算法思想：

树状数组的写法是要建立一个c数组

如果i是奇数c[i]=a[i]:c[3]=a[3].如果i是偶数：c[2]=a[1]+a[2],c[4]=a[1]+a[2]+a[3]+a[4],c[6]=a[5]+a[6]
如果i的二进制最后一个1是几就是往前几项和：6（110）最后一个1代表2所以c[6]=a[5]+a[6],7(111)代表1所以c[7]=a[7],8(1000)代表8所以c[8]=a[1]......+a[8];

lowbit
{
return x-(x&(x-1);
}

奇数（x&（x-1））一定是x-1，如111和110与还是110，x-（x-1）还是1所以都是一项

偶数 如6 ，110和101与肯定是100，x-100=10所以就是2项

构造树状数组就是更新操作
sum[7]=c[4]+c[6]+c[7]前7项和   sum[111]=c[100]+c[110]+c[111]求前i项一定要包括c[i]

int sum(i)//求前i项所有的和
{
int ret =0;
     while(i>0)
   {
     ret+=c[i];//从右往左求和
     i-=lowbit(i);
   }
return   ret;
}
时间复杂度O（logn）
更新也为logn
void up(int i,int val)
{
     while(i<=n)
 {
    c[i]+=val;
    i+=lowbit(i);
 }

}
 

最终代码

 #include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=1e7+10;
int a[N],c[N];
int n,m;
int lowbit(int x)
{
    return x&-x;//
}
void up(int x,int y)
{
    while(x<=n)
    {
        c[x]+=y;
        x+=lowbit(x);
    }
}
int sum(int i)
{
    int ret=0;
    while(i>0)
    {
        ret+=c[i];
        i-=lowbit(i);
    }
    return ret;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
    cin>>a[i];
    int q=i;
    up(i,a[i]);
//for(int z=i;z>i-lowbit(i);z--)
//c[i]+=a[z];
    
}
   for(int i=1;i<=m;i++)
   {
       int x,y,z;
       cin>>z>>x>>y;
       if(z==1)
       {
       up(x,y);
           
       }
    else if(z==2)
    {
        cout<<sum(y)-sum(x-1)<<endl;
    }
       
       
   }
    
}