树状数组(单点修改,区间查询)

题目描述

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

1.将某一个数加上x

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

输入输出格式

输入格式:

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

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

接下来M行每行包含3个整数,表示一个操作,具体如下:

操作1: 格式:1 x k 含义:将第x个数加上k

操作2: 格式:2 x y 含义:输出区间[x,y]内每个数的和

输出格式:

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

#include<bits/stdc++.h>
using namespace std;
long long a[500009];
long long b[500009];
int n,m;
int lowbit(int x)
{
    return (x&(-x));
}

void change(int x,int p)
{
    for(int i=p;i<=n;i+=lowbit(i))
    {
        b[i]+=x;
    }
}
long long query(int p)
{
    long long sum=0;
    for(int i=p;i>=1;i-=lowbit(i))
    {
        sum+=b[i];
    }
    return sum;
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<=n;i++)
    {
        for(int j=i;j<=n;j+=lowbit(j))
        {
            b[j]+=a[i];
        }
    }
    for(int i=1;i<=m;i++)
    {
        int o;
        cin>>o;
        int x,y;
        cin>>x>>y;
        if(o==1)change(y,x);
        if(o==2)cout<<query(y)-query(x-1)<<endl;
    }            
}

猜你喜欢

转载自www.cnblogs.com/1129-tangqiyuan/p/11220993.html