线段树 模板

#define MAXSIZE 50010

int tree[4*MAXSIZE];　　// 此处要开4倍空间
int lz[4*MAXSIZE];

void init()
{
    memset(tree, 0, sizeof(tree));
    memset(lz, 0, sizeof(lz));
} 


void build(int node, int l, int r)
{
    if(l == r)    // 到达叶子节点，赋值
    {
        scanf("%d", &tree[node]);
        return;
    } 
    
    int mid = (l+r)/2;
    build(node*2, l, mid);
    build(node*2+1, mid+1, r);
    
    tree[node] = tree[node*2] + tree[node*2+1];
}


void push_down(int node, int l, int r)
{
    if(lz[node])
    {
        int mid = (l+r)/2;
        lz[node*2] += lz[node];
        lz[node*2+1] += lz[node];
        tree[node*2] += (mid-l+1)*lz[node];
        tree[node*2+1] += (r-mid)*lz[node];
        lz[node] = 0;
    }
}

// 单点更新，node为1(从根开始)，lr为根的范围，index为待更新的数，add为更新的增量 
void update(int node, int l, int r, int index, int add)
{
    if(l == r)
    {
        tree[node] += add;    // 更新方式 
        return;
    }
    
    int mid = (l+r)/2;
    if(index <= mid)    // 进入左子树 
        update(node*2, l, mid, index, add);
    else                // 进入右子树 
        update(node*2+1, mid+1, r, index, add);
        
    tree[node] = tree[node*2] + tree[node*2 + 1]; 
}

// 区间更新，lr为更新范围，LR为线段树范围，add为更新值
void update_range(int node, int l, int r, int L, int R, int add)
{
    if(l <= L && r >= R)
    {
        lz[node] += add;
        tree[node] += (R-L+1)*add;    // 更新方式 
        return;
    }
    
    push_down(node, L, R);
    int mid = (L+R)/2;
    if(mid >= l)    // 进入左子树 
        update_range(node*2, l, r, L, mid, add);
    if(r > mid)        // 进入右子树 
        update_range(node*2+1, l, r, mid+1, R, add);
        
    tree[node] = tree[node*2] + tree[node*2 + 1]; 
}

// 区间查找 
int query_range(int node, int l, int r, int L, int R)
{
    if(l <= L && r >= R)
        return tree[node];
    push_down(node, L, R);
    int mid = (L+R)/2;
    int sum = 0;
    if(mid >= l)
        sum += query_range(node*2, l, r, L, mid);
    if(r > mid)
        sum += query_range(node*2+1, l, r, mid+1, R);
    
    return sum;
}