P3374 【模板】树状数组 1

Lzy1930

于 2024-01-17 14:03:08 发布

阅读量985

点赞数 25

文章标签：算法数据结构

本文链接：https://blog.csdn.net/Li1502927435/article/details/135648934

版权

【模板】树状数组 1

题目描述

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

将某一个数加上 $x$
求出某区间每一个数的和

输入格式

第一行包含两个正整数 $n, m$ ，分别表示该数列数字的个数和操作的总个数。

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

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

1 x k 含义：将第 $x$ 个数加上 $k$
2 x y 含义：输出区间 $[x, y]$ 内每个数的和

输出格式

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

样例 #1

样例输入 #1

样例输出 #1

14
16

提示

【数据范围】

对于 $30\%$ 的数据， $\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);//求最后的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]);//求c数组
//for(int z=i;z>i-lowbit(i);z--)求c数组
//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;
    }
       
       
   }
    
}