线段树：Segment Tree(单点修改/区间修改模板) C++

线段树是非常有效的数据结构，可以快速的维护单点修改，区域修改，查询最大值，最小值等功能。

同时，它也很重要。如果有一天比赛，你卡在了一道线段树模板题目上，这就真的尴尬了。不过，随着时代的进步，题目也越来越变态，线段树更多时候则是你算法时间复杂度的优化。

这是单点查询的代码。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
 
int n;
int a[];
 
struct node
{
    int l;
    int r;
    int sum;
} t[];
 
void build (int i,int l,int r)
{
    t[i].l=l;
    t[i].r=r;
    int mid=l+r>>;
    if (l==r)
    {
        t[i].sum=a[l];return ;
    }
 
    build(i<<,l,mid);
    build((i<<)+,mid+,r);
 
    t[i].sum=t[i<<].sum+t[(i<<)+].sum;
 
    return ;
}
 
void update(int i,int p,int k)
{
    if (t[i].l==t[i].r)
    {
        t[i].sum=k;
        return;
    }
 
    int mid=t[i].l+t[i].r>>;
    if (p<=mid) update(i<<,p,k);
    if (p>mid) update((i<<)+,p,k);
    t[i].sum=t[i<<].sum+t[(i<<)+].sum;
 
    return ;
}
 
int getsum(int i,int l,int r)
{
    int mid=(t[i].l+t[i].r)>>;
    if (l<=t[i].l&&t[i].r<=r) return t[i].sum;
    if (r<=mid) return getsum(i<<,l,r);
    if (l>mid) return getsum((i<<)+,l,r);
 
    return getsum(i<<,l,r)+getsum((i<<)+,l,r);
}
 
int main()
{
    int m;
    scanf("%d%d",&n,&m);
    for (int i=;i<=m;i++)
    {
        scanf("%d",&a[i]);
    }
    build(,,n);
    while (m--)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        if (x==)
        {
            update(,y,z);
        }
        else printf("%d\n",getsum(,y,z));
    }
 
    return ;
}

下面是区间修改的代码。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cmath>
#include <set>
using namespace std;
 
struct node
{
    int l,r,s,d;
} t[];
int n,m;
int a[];
 
void pushup(int x)
{
    t[x].s=t[x<<].s+t[(x<<)+].s;
    return;
}
 
void pushdown(int x)
{
    int mid=(t[x].l+t[x].r)>>;
    int d=t[x].d;
    t[x<<].d+=d;t[(x<<)+].d+=d;
    t[x<<].s+=(mid-t[x].l+)*d;
    t[(x<<)+].s+=(t[x].r-mid)*d;
    return;
}
 
void build(int i,int l,int r)
{
    t[i].l=l;
    t[i].r=r;
    t[i].d=;
    if (l==r)
    {
        t[i].s=a[l];
        return ;
    }
 
    int mid=l+r>>;
 
    build(i<<,l,mid);
    build((i<<)+,mid+,r);
 
    pushup(i);
    return;
}
 
int getsum(int i,int l,int r)
{
    int mid=t[i].l+t[i].r>>;
    if (l<=t[i].l&&t[i].r<=r)
    {
        return t[i].s;
    }
    int d=t[i].d;
    if (!d) pushdown(i);
    int ret=;
    if (mid>=l) ret+=getsum(i<<,l,r);
    if (mid<r) ret+=getsum((i<<)+,l,r);
 
    return ret;
}
 
void update(int i,int l,int r,int del)
{
    if (t[i].l==t[i].r)
    {
        t[i].d+=del;
        t[i].s+=(t[i].l-t[i].r+)*del;
        return ;
    }
 
    int mid=t[i].l+t[i].r>>;
    if (!t[i].d) pushdown(i);
    if (mid>=l) update(i<<,l,r,del);
    if (mid<r) update((i<<)+,l,r,del);
    pushup(i);
}
 
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    build (,,n);
 
    for (int i=;i<=m;i++)
    {
        int x,y,z,d;
        scanf("%d",&x);
        if (x==)
        {
            scanf("%d%d%d",&y,&z,&d);
            update(,y,z,d);
        }
        if (x==)
        {
            scanf("%d%d",&y,&z);
            printf("%d\n",getsum(,y,z));
        }
    }
 
    return ;
}