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

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


一.单点更新

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


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

#include
#include
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=right) add(left,right,i*2,k);
    else if(mid

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

#include
#include
#include
#include
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);
        }
    }
}

2.(HDU1754) 也是单点更新,但是不是求区间和,而是求区间最大值,和上面一样,只需要拿求和操作改为取最大值操作


上代码:

#include 
#include 
#include 
#include 
#include 
#include
#pragma comment(linker,"/STACK:1024000000,1024000000")
using namespace std;
const int maxn = 2e5+5;
int maxz[maxn*4];
int num[maxn];
void up(int rt){
    maxz[rt]=max(maxz[rt<<1],maxz[rt<<1|1]);
}
void build(int l,int r,int rt){
    if(l==r){
        maxz[rt]=num[l];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    up(rt);
}
void update(int p,int z,int l,int r,int rt){
    if(l==r){
        maxz[rt]=z;
        return;
    }
    int mid=(l+r)>>1;
    if(p<=mid) update(p,z,l,mid,rt<<1);
    else update(p,z,mid+1,r,rt<<1|1);
    up(rt);
}
int query(int L,int R,int l,int r,int rt){
    if(L<=l&&r<=R){
        return maxz[rt];
    }
    int mid=(l+r)>>1;
    int ret=0;
    if(L<=mid) ret=max(ret,query(L,R,l,mid,rt<<1));
    if(R>mid) ret=max(ret,query(L,R,mid+1,r,rt<<1|1));
    return ret;
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
        memset(maxz,0,sizeof(maxz));
        for(int i=1;i<=n;i++) scanf("%d",&num[i]);
        build(1,n,1);
        while(m--){
            char c;
            int x,y;
            scanf(" %c%d%d",&c,&x,&y);
            if(c=='U') update(x,y,1,n,1);
            if(c=='Q') printf("%d\n",query(x,y,1,n,1));
        }
    }
}



3.(HDU2795 billboard)题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=2795

题意:给一块h*w广告板,然后给n个1*wi的广告条,广告条放的顺序是有限选择向上的,再优先选择左边的,对于每块广告条,输出它放的位置,如果放不下,输出-1


还是一个求max的单点更新,只不过换了个外套,注意线段树开的范围就好了


题解链接:http://blog.csdn.net/sin_xf/article/details/47398289



二、区间更新

区间相比于单点更新,多了个懒惰标记,每次更新或者求和的时候遇到懒惰区间要吧懒惰标记网下传


1.先上个模板题(POJ3468 A Simple Problem with Integers)

就是线段树区间求和的模板题,注意数组和结果开long long就好了

直接上题解:http://blog.csdn.net/sin_xf/article/details/47401051


2.(HDU1698 Just a Hook)http://acm.hdu.edu.cn/showproblem.php?pid=1698

m次操作,每次操作把一段区间每个点的值给替换,成段替换,区间求和的变形

题解:http://blog.csdn.net/sin_xf/article/details/47402267


3.(ZOJ1610 Count the Colors)http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=610

给一个n,代表n次操作,接下来每次操作表示把[l,r]区间的线段涂成k的颜色其中,l,r,k的范围都是0到8000

拿线段树维护一段区间的颜色,区间更新,但是最后要延迟标记更新到底,相当与n次更新,一次询问

题解:http://blog.csdn.net/sin_xf/article/details/47726363


三、区间合并

1、先上个经典的区间合并题(Hotel POJ3667)

大概就是给n个点,m次操作,操作类型两种,一种求连续区间为d的最左边的点,并把这段区间覆盖,第二种把一段区间回收

题解:http://blog.csdn.net/sin_xf/article/details/47681763




你可能感兴趣的:(线段树)