树状数组

最新推荐文章于 2021-07-03 15:09:04 发布

Healer66

最新推荐文章于 2021-07-03 15:09:04 发布

阅读量150

点赞数

分类专栏： ACM知识点

本文链接：https://blog.csdn.net/Healer66/article/details/83045756

版权

ACM知识点专栏收录该内容

46 篇文章 3 订阅

订阅专栏

https://blog.csdn.net/Small_Orange_glory/article/details/81290634

#include<bits/stdc++.h>
#define MAXN 9999
using namespace std;
string op;//指令
int a[MAXN];//原始数组
int c[MAXN];//树状数组
int n;//原始数组的大小
int q;//查询前缀和时查询从数组1到q位置上的和
int z,l;//更新后缀和时原始数组z位置变化了l
int lowbit(int n)//假如说n等于8，那么8的2进制为：1000。函数返回的是2^3（二进制中有从后往前连续的3个0），即：8。
{
    return n&(-n);
}
int sum(int q)//求前缀和（a数组n位置到1位置上的数的和）
{
    int ret=0;
    for(int i=q;i>=1;i-=lowbit(i))
        ret+=c[i];
    return ret;
}
void update(int i,int j)//当a数组i位置上的数变化了j后，重新更新树状数组c
{
    for(int k=i;k<=n;k+=lowbit(k))
        c[k]+=j;
}
int main()
{
    while (cin>>n)
    {
        memset(c,0,sizeof(c));
        memset(a,0,sizeof(a));
        for(int i=1;i<=n;i++)
        {
            cin>>a[i];
            update(i,a[i]);//刚开始建立树状数组时，每次都得进行这个过程，和建立大根堆（小根堆）时差不多，每次都得进行heapinsert过程
        }
        cin>>op;
        if(op=="update")
        {
            cin>>z>>l;
            update(z,l);
        }
        if(op=="sum")
        {
            cin>>q;
            cout<<sum(q)<<endl;
        }
    }

}