洛谷-P3374-树状数组模板题

题目链接
题意:
给你一个数组,对这个数组进行单点修改或区间求和.
思路:
树状数组模板
代码:

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=1e6+7;
const int inf=0x3f3f3f3f;
int n,m,tree[2000005];
int lowbit(int k)
{
    return (k)&(-k);
}
void add(int x,int k)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        tree[i]+=k;
    }
}
int getsum(int x)
{
    int ans=0;
    for(int i=x;i;i-=lowbit(i))
    {
        ans+=tree[i];
    }
    return ans;
}
signed main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        int a;
        cin>>a;
        add(i,a);
    }
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        if(a==1)
        {
            add(b,c);
        }
        if(a==2)
        {
            cout<<getsum(c)-getsum(b-1)<<endl;
        }
    }
}

猜你喜欢

转载自blog.csdn.net/ACkingdom/article/details/106461794