线段树模板

#include <cstdio>
#include <algorithm>
#include <iostream>
using namespace std;
int n,m,p;//n个结点，m种操作，操作为p
int x,y;//查询的值x或者单点修改x的值加上y
int a,b;//区间查询时使用的区间a,b;
int ans;//存储答案
struct node
{
    int left,right;//左右
    int weight;//权值
    int flag;//延迟标记
} tree[400001];
void build(int k,int ll,int rr)//建树，k是父亲节点，ll、rr左右孩子
{
    tree[k].left=ll,tree[k].right=rr;
    if(tree[k].left==tree[k].right)//如果是叶子结点，直接赋值
    {
        scanf("%d",&tree[k].weight);
        return;
    }
    int m=(ll+rr)/2;
    build(k*2,ll,m);
    build(k*2+1,m+1,rr);
    tree[k].weight=tree[k*2].weight+tree[k*2+1].weight;//求区间和，所以父亲结点是左右孩子的和
}
void down(int k)//延迟标记下传
{
    tree[k*2].flag+=tree[k].flag;
    tree[k*2+1].flag+=tree[k].flag;
    tree[k*2].weight+=tree[k].flag*(tree[k*2].right-tree[k*2].left+1);
    tree[k*2+1].weight+=tree[k].flag*(tree[k*2+1].right-tree[k*2+1].left+1);
    tree[k].flag=0;
}
void ask_point(int k)//单点查询
{
    if(tree[k].left==tree[k].right)
    {
        ans=tree[k].weight;
        return ;
    }
    if(tree[k].flag) down(k);
    int m=(tree[k].left+tree[k].right)/2;
    if(x<=m) ask_point(k*2);
    else ask_point(k*2+1);
}
void change_point(int k)//单点修改，第x个数加上y
{
    if(tree[k].left==tree[k].right)
    {
        tree[k].weight+=y;
        return;
    }
    if(tree[k].flag) down(k);
    int m=(tree[k].left+tree[k].right)/2;
    if(x<=m) change_point(k*2);
    else change_point(k*2+1);
    tree[k].weight=tree[k*2].weight+tree[k*2+1].weight;
}
void ask_interval(int k)//区间查询，查询区间和
{
    if(tree[k].left>=a&&tree[k].right<=b)
    {
        ans+=tree[k].weight;
        return;
    }
    if(tree[k].flag) down(k);
    int m=(tree[k].left+tree[k].right)/2;
    if(a<=m) ask_interval(k*2);
    if(b>m) ask_interval(k*2+1);
}
void change_interval(int k)//区间修改，把a-b之间数改成k
{
    if(tree[k].left>=a&&tree[k].right<=b)
    {
        tree[k].weight+=(tree[k].right-tree[k].left+1)*y;
        tree[k].flag+=y;
        return;
    }
    if(tree[k].flag) down(k);
    int m=(tree[k].left+tree[k].right)/2;
    if(a<=m) change_interval(k*2);
    if(b>m) change_interval(k*2+1);
    tree[k].weight=tree[k*2].weight+tree[k*2+1].weight;
}
int main()
{
    scanf("%d",&n);//n个节点
    build(1,1,n);//建树，传值根节点1，左孩子1，有孩子n
    scanf("%d",&m);//m种操作
    for(int i=1; i<=m; i++)
    {
        scanf("%d",&p);
        ans=0;
        /*以下操作传值过去的为根节点1*/
        if(p==1)
        {
            scanf("%d",&x);
            ask_point(1);//单点查询,输出第x个数
            printf("%d",ans);
        }
        else if(p==2)
        {
            scanf("%d%d",&x,&y);
            change_point(1);//单点修改
        }
        else if(p==3)
        {
            scanf("%d%d",&a,&b);//区间查询
            ask_interval(1);
            printf("%d\n",ans);
        }
        else
        {
            scanf("%d%d%d",&a,&b,&y);//区间修改
            change_interval(1);
        }
    }
}