线段树模板及专题合集-----不断更新中

标签：

之前学过一段时间的线段树，后来就没有再做了，上次比赛做到几道题居然都是线段树，而且渐渐发现线段树的适用性之广和重要性，而且之前学的不全面，于是决定重现学一遍线段树，顺便改一下之前的代码风格

一.单点更新

最典型的最基础的题目就是敌兵布阵了，题意大概是给n个数，然后3种操作：某个点的数加上x，某个点的数减去x，求[x，y]区间所有数的和。前两种操作都一样，只不过减去x相当于加上-x

先贴一段以前写的搓代码：

#include<stdio.h>
#include<string.h>
const int MAXNODE = 50005;
typedef struct NODE{
    int left,right;
    int num;
}Node;
int ans=0;
int num[50010];
Node node[140000];
void creattree(int i,int left,int right)
{
    node[i].left=left;
    node[i].right=right;
    if(left==right)
    {
        node[i].num=num[right];
        return;
    }
    creattree(i*2,left,(left+right)/2);
    creattree(i*2+1,(left+right)/2+1,right);
    node[i].num+=node[i*2].num+node[i*2+1].num;
}
void sum(int left,int right,int i)
{
    if(node[i].left==left&&node[i].right==right)
    {
        ans+=node[i].num;
        return;
    }
    else{
        int mid=(node[i].left+node[i].right)/2;
        if(mid>=right) sum(left,right,i*2);
        else if(mid<left) sum(left,right,i*2+1);
        else{
            sum(left,mid,i*2);
            sum(mid+1,right,i*2+1);
        }
    }
}
void add(int left,int right,int i,int k)
{
    node[i].num+=k;
    if(node[i].left==left&&node[i].right==right)
    {
        return;
    }
    else{
    int mid=(node[i].left+node[i].right)/2;
    if(mid>=right) add(left,right,i*2,k);
    else if(mid<left) add(left,right,i*2+1,k);}
}
int main()
{
    char s[10];
    int t;
    while(scanf("%d",&t)!=EOF)
    {
        for(int Case=1;Case<=t;Case++)
        {
            memset(num,0,sizeof(0));
            int n;
            scanf("%d",&n);
            for(int i=1;i<=n;i++)
            {
                scanf("%d",&num[i]);
            }
            creattree(1,1,n);
            printf("Case %d:\n",Case);
            while( scanf("%s",s))
            {
                int a,b;

                if(s[0]=='E') break;
                scanf("%d%d",&a,&b);
                if(s[0]=='A') add(a,a,1,b);
                else if(s[0]=='S') add(a,a,1,-b);
                else if(s[0]=='Q')
                {
                    ans=0;
                    sum(a,b,1);
                    printf("%d\n",ans);
                }
            }
        }
    }
    return 0;
}

接着是最新更新的代码风格：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn = 5e5+5;
int sum[maxn*4];
int num[maxn];
void up(int rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int rt,int l,int r){
    if(l==r) {sum[rt]=num[l];return;}
    int m=(l+r)>>1;
    build(rt<<1,l,m);
    build(rt<<1|1,m+1,r);
    up(rt);
}
void update(int p,int add,int rt,int l,int r){
    if(l==r) {
        sum[rt]+=add;
        return;
    }
    int m=(l+r)>>1;
    if(p>m) update(p,add,rt<<1|1,m+1,r);
    else update(p,add,rt<<1,l,m);
    up(rt);
}
int query(int L,int R,int rt,int l,int r){
    if(l>=L&&r<=R) return sum[rt];
    int m=(l+r)>>1;
    int ret = 0;
    if(L<=m) ret+=query(L,R,rt<<1,l,m);
    if(R>m) ret+=query(L,R,rt<<1|1,m+1,r);
    return ret;
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int cas=1;cas<=t;cas++){
        int n;
        memset(sum,0,sizeof(sum));
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d",&num[i]);
        }
        build(1,1,n);
        printf("Case %d:\n",cas);
        char s[10];
        while(scanf("%s",s)!=EOF&&strcmp(s,"End")){
            int x,y;
            scanf("%d%d",&x,&y);
            if(s[0]=='Q') printf("%d\n",query(x,y,1,1,n));
            else if(s[0]=='A') update(x,y,1,1,n);
            else if(s[0]=='S') update(x,-y,1,1,n);
        }
    }
}

线段树模板及专题合集-----不断更新中

标签：

原文地址：http://blog.csdn.net/sin_xf/article/details/47146359