线段树学习笔记

线段树的原理

四个基本操作

例题一：
给定 n 个数组成的一个数列，规定有两种操作，一是修改某个元素，二是求子数列 [a,b] 的连续和。

输入格式
第一行包含两个整数 n 和 m，分别表示数的个数和操作次数。

第二行包含 n 个整数，表示完整数列。

接下来 m 行，每行包含三个整数 k,a,b （k=0，表示求子数列[a,b]的和；k=1，表示第 a 个数加 b）。

数列从 1 开始计数。

输出格式
输出若干行数字，表示 k=0 时，对应的子数列 [a,b] 的连续和。

代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>

using namespace std;

const int N=100010;

struct Node
{
    int l,r;
    int sum;
}tr[N*4];
int w[N];
int n,m;

void pushup(int u)
{
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}

void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r,w[r]};
    else 
    {
        tr[u]={l,r};
        int mid=tr[u].l+tr[u].r>>1;
        build(u<<1,l,mid),build(u<<1|1,mid+1,r);
        pushup(u);
    }
}

int query(int u,int l,int r)
{
        if(tr[u].l>=l && tr[u].r<=r) return tr[u].sum;
    
        int mid=tr[u].l+tr[u].r>>1;
        int sum=0;
        if(l<=mid) sum+=query(u<<1,l,r);
        if(r>mid) sum+=query(u<<1|1,l,r);

        return sum;
}

void modify(int u,int a,int b)
{
    if(tr[u].l==tr[u].r) tr[u].sum+=b;
    else
    {
        int mid=tr[u].l+tr[u].r>>1;
        if(a<=mid) modify(u<<1,a,b);
        else 
            modify(u<<1|1,a,b);
        pushup(u);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    build(1,1,n);
    
    while(m--)
    {
        int k,a,b;
        scanf("%d%d%d",&k,&a,&b);
        if(k==0)
        {
            cout<<query(1,a,b)<<endl;
        }
        else
        {
            modify(1,a,b);
        }
    }
    
    return 0;
}