洛谷 P3374 【模板】树状数组 1 题解

题目描述

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

• 将某一个数加上 x
  

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

输入格式

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

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

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

• 1 x k 含义：将第 x 个数加上 k
  

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

输出格式

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

AC代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;

ll n,m,a[500001],c[500001];

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

void updata(int k,int v)
{
    while (k<=n)
    {
        c[k]+=v;
        k+=lowbit(k);
    }
}

int getsum(int k)
{
    int ans=0;
    while (k>0)
    {
        ans+=c[k];
        k-=lowbit(k);
    }
    return ans;
}

int main()
{
    scanf("%lld%lld",&n,&m);
    for (int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        updata(i,a[i]);
    }
    while (m--)
    {
        int op,x,y;
        cin>>op>>x>>y;
        if (op==1)
            updata(x,y);
        if (op==2)
        {
            int sum=getsum(y)-getsum(x-1);
            printf("%d\n",sum);
        }
    }
    return 0;
}