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

传送门

　　

---

 

　　题目描述：

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

　　1.将某一个数加上x

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

---

 

　　这道题看名字就知道是树状数组的模板题，但是菜鸡的我想熟悉线段树，所以用了线段树。

　　思路没有什么好说的(如果不知道怎么做的，建议先学习树状数组)，详情看代码。

　　一些小细节：

　　　　1.运算都用位运算加速

　　　　2.不要用cin,cout，亲测超时

　　　　3.对于根node，

　　　　　　左儿子 是 node * 2       ---->> node << 1;

　　　　　　右儿子是  node * 2 + 1 ---->> node<<1 | 1;

#include <iostream>
#include <cstdio>
using namespace std;
const int N = 5e5 + 10;

int n, m;
int tree[N*4];


//不用建树，因为你会发现代码是一样的。。。亲测 

//在pos上加上X 
void update(int node, int l, int r, int pos, int x)
{
    if(l == r)    {tree[node] += x; return;}
    int mid = (l + r) >> 1;
    if(mid >= pos)    update(node<<1, l, mid, pos, x);
    else            update((node<<1)|1, mid+1, r, pos, x);
    tree[node] = tree[node<<1] + tree[(node<<1)+1]; 
}

//查询区间[L, R]的和 
int query(int node, int l, int r, int L, int R)
{
    if(L > r || R < l)        return 0;
    if(L <= l && R >= r)    return tree[node];
    int mid = (l + r) >> 1;
    return query(node<<1, l, mid, L, R) + query((node<<1)|1, mid+1, r, L, R);
}

int main()
{
    scanf("%d %d", &n, &m);
    for(int i = 1; i <= n; i++)    
    {
        int x;
        scanf("%d", &x);
        update(1, 1, N, i, x);
    }
    for(int i = 0; i < m; i++)
    {
        int op, x, y;
        scanf("%d %d %d", &op, &x, &y);
        if(op == 1)    update(1, 1, N, x, y);
        else        printf("%d\n", query(1, 1, N, x, y));
    }
    return 0;
}