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

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

---

题目

题目描述

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

1.将某一个数加上x

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

输入输出格式

输入格式：
第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。

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

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

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

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

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

输入输出样例

输入样例#1：

5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4

输出样例#1：

14
16

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=10000，M<=10000

对于100%的数据：N<=500000，M<=500000

---

题解

树状数组（最不想打的东西，感觉自己理解有点难QAQ）

关键操作lowbit($x$&-$x$) 就是找到二进制下x的最后一个1的位置

（-$x$在二进制下的表达就是对$x$按位取反后+1） ps：我也不确定是不是……

（不知道的话最好自己打一个代码，找找规律，应该也可以看懂这个操作）

然后就是根据这个性质做一个类似线段树的东西（具体的我也说不来，反正加、乘等操作就是不断将子节点的信息更新到父节点，然后询问操作就是从父节点分解到各个子节点）

---

代码

#include<cstdio>
using namespace std;

int n,m;
int a[500005],sum[500005];

int readln()
{
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=getchar();}
    while ('0'<=ch&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x*f;
}

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

void add(int x,int y)
{
    while (x<=n)
    {
        sum[x]+=y;x+=lowbit(x);
    }
}

int get(int x) 
{
    int ret=0;
    while (x>0)
    {
        ret+=sum[x];
        x-=lowbit(x);
    }
    return ret;
}

int main()
{
    n=readln();m=readln();
    for (int i=1;i<=n;i++) add(i,readln());
    for (int i=1;i<=m;i++)
    {
        int x=readln(),y;
        if (x==1) {
            x=readln();y=readln();
            add(x,y);
        }
        else {
            x=readln();y=readln();
            printf("%d\n",get(y)-get(x-1));
        }
    }
    return 0;
}