树状数组模板

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

题目链接：https://www.luogu.org/problemnew/show/P3374

AC代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
int n,m,tree[2000010];
int lowbit(int k)
{
    return k & -k;
}
void add(int x,int k)
{
    while(x<=n)
    {
        tree[x]+=k;
        x+=lowbit(x);
    }
}
int sum(int x)
{
    int ans=0;
    while(x!=0)
    {
        ans+=tree[x];
        x-=lowbit(x);
    }
    return ans;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        int a;
        scanf("%d",&a);
        add(i,a);
    }
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        if(a==1)
            add(b,c);                        //将第b个数加上c
        if(a==2)
            cout<<sum(c)-sum(b-1)<<endl;     //输出区间[b,c]中每个数的和
    }
}

P3368 【模板】树状数组 2

题目链接：https://www.luogu.org/problemnew/show/P3368

AC代码

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
using namespace std;
int n,m;
int input[500010];
int tree[500100];
int lowbit(int x)
{
    return x & -x;
}
void add(int x,int k)
{
    while(x<=n)
    {
        tree[x]+=k;
        x+=lowbit(x);
    }
}
int search(int x)
{
    int ans=0;
    while(x!=0)
    {
        ans+=tree[x];
        x-=lowbit(x);
    }
    return ans;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>input[i];
    for(int i=1;i<=m;i++)
    {
        int a;
        scanf("%d",&a);
        if(a==1)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            //将区间[x,y]内每个数加上z 
            add(x,z);
            add(y+1,-z);
        }
        if(a==2)
        {
            int x;
            scanf("%d",&x);
            //输出第x个数的值 
            printf("%d\n",input[x]+search(x));
        }
    }
}

以上代码我是直接copy的，没有看懂树状数组的原理，只是总结一个模板，侵权删。