线段树模板

     今天看了半天，主要是知识点，还看了些博客，最后总结出来线段树的这么一套模板，连带着自己加的些理解注释，模板库终于不是空的了。

     ！！！补！！！后来用模板超时，发现错误点，应该在查询时加上一步判断——是否 tree[].r<left || tree[].l>right，是则return 。（已在代码中加上）。

#include<iostream>
using namespace std;
const int MAX=10010;
int N,M,ans;
int a[MAX];
struct node
{
    int l,r;
    int sum;
    int f;
}tree[MAX*4+1];
inline void build(int k,int left,int right)//建树
{
    tree[k].l=left;
    tree[k].r=right;
    if(tree[k].l==tree[k].r)//如果是子节点
    {
        tree[k].sum=a[left];
        return ;
    }
    int mid=(left+right)/2;
    build(k*2,left,mid);//左子树
    build(k*2+1,mid+1,right);//右子树
    tree[k].sum=tree[k*2].sum+tree[k*2+1].sum;//回溯  修改当前
}
inline void down(int k)//标记下传
{
    tree[k*2].f+=tree[k].f;//传向左右子树
    tree[k*2+1].f+=tree[k].f;
    tree[k*2].sum+=tree[k].f*(tree[k*2].r-tree[k*2].l+1);
    tree[k*2+1].sum+=tree[k].f*(tree[k*2+1].r-tree[k*2+1].l+1);
    tree[k].f=0;//本身设为 0
}
inline void queryPoint(int k,int pos)//单点查询
{//k是当前节点    pos是要查询数组位置
    if(tree[k].l==tree[k].r)
    {
        ans=tree[k].sum;
        return ;
    }
    if(tree[k].f)//如果有标记  下传
        down(k);
    int mid=(tree[k].l+tree[k].r)/2;
    if(pos<=mid)
        queryPoint(k*2,pos);
    else
        queryPoint(k*2+1,pos);
}
inline void changePoint(int k,int pos,int d)//单点修改
{//k是当前位置   pos是数组中位置   d是要求改成的数
    if(tree[k].l==tree[k].r)
    {
        tree[k].sum+=d;//不是修改  是 +d
        return;
    }
    if(tree[k].f)//下传
        down(k);
    int mid=(tree[k].l+tree[k].r)/2;
    if(pos<=mid)
        changePoint(k*2,pos,d);
    else
        changePoint(k*2+1,pos,d);
    tree[k].sum=tree[k*2].sum+tree[k*2+1].sum;
}
inline void queryInterval(int k,int left,int right)//区间查询
{
    if(tree[k].r<left || tree[k].l>right)//后来补错加上的，否则超时
        return ;
    if(tree[k].l>=left && tree[k].r<=right)
    {//当节点k的（l，r）在【left，right】中间时  ans+
        ans+=tree[k].sum;
        return;
    }
    if(tree[k].f)//下传
        down(k);
    int mid=(tree[k].l+tree[k].r)/2;
    if(left<=mid)
        queryInterval(k*2,left,right);
    if(right>mid)
        queryInterval(k*2+1,left,right);
}
inline void changeInterval(int k,int left,int right,int d)//区间修改
{
    if(tree[k].l>=left && tree[k].r<=right)//节点k完全被包含时
    {
        tree[k].sum+=(tree[k].r-tree[k].l+1)*d;
        tree[k].f+=d;//不是修改  是 +d
        return;
    }
    if(tree[k].f)
        down(k);
    int mid=(tree[k].l+tree[k].r)/2;
    if(left<=mid)
        changeInterval(k*2,left,right,d);
    if(right>mid)
        changeInterval(k*2+1,left,right,d);
    tree[k].sum=tree[k*2].sum+tree[k*2+1].sum;//回溯
}
int main()
{
    int p,x,l,r,d;
    cin>>N;//n个节点
    for(int i=1; i<=N; i++)
        cin>>a[i];
    build(1,1,N);//建树
    cout<<"建树ok"<<endl;
    cin>>M;//m种操作
    for(int i=1;i<=M;i++)
    {
        cout<<"选择操作："<<endl;
        cin>>p;
        ans=0;
        if(p==1)
        {
            cin>>x;
            queryPoint(1,x);//单点查询,输出第x个数
            cout<<ans<<endl;;
        }
        else if(p==2)
        {
            cin>>x>>d;
            changePoint(1,x,d);//单点修改
        }
        else if(p==3)
        {
            cin>>l>>r;
            queryInterval(1,l,r);//区间查询
            cout<<ans<<endl;
        }
        else
        {
            cin>>l>>r>>d;
            changeInterval(1,l,r,d);
        }
    }
    return 0;
}