线段树（区间）

线段树总结

模板

特殊声明

#definr SIZE 
#define myq l,m,node<<1
#define wbh m+1,r,node<<1|1
struct Node{
    int sum;
    int len;
    int lazy;
    int lmax;
    int rmax;
}myq[SIZE<<2];

lazy标记

一般步骤
查找
- 判断区间是否达到最小
- 将lazy更新向下传递
- 递归查询
更新
- 判断区间是否达到最小
- 将lazy更新向下传递
- 递归更新
- 将父亲节点特征变量更新

例题解析

【Usaco2008 Feb】Hotel 旅馆
题目描述
奶牛们最近的旅游计划，是到苏必利尔湖畔，享受那里的湖光山色，以及明媚的阳光。作为整个旅游的策划者和负责人，贝茜选择在湖边的一家著名的旅馆住宿。这个巨大的旅馆一共有N (1 <= N <= 50,000)间客房，它们在同一层楼中顺次一字排开，在任何一个房间里，只需要拉开窗帘，就能见到波光粼粼的湖面。 贝茜一行，以及其他慕名而来的旅游者，都是一批批地来到旅馆的服务台，希望能订到D_i (1 <= D_i <= N)间连续的房间。服务台的接待工作也很简单：如果存在r满足编号为r..r+D_i-1的房间均空着，他就将这一批顾客安排到这些房间入住；如果没有满足条件的r，他会道歉说没有足够的空房间，请顾客们另找一家宾馆。如果有多个满足条件的r，服务员会选择其中最小的一个。 旅馆中的退房服务也是批量进行的。每一个退房请求由2个数字X_i、D_i 描述，表示编号为X_i..X_i+D_i-1 (1 <= X_i <= N-D_i+1)房间中的客人全部离开。退房前，请求退掉的房间中的一些，甚至是所有，可能本来就无人入住。 而你的工作，就是写一个程序，帮服务员为旅客安排房间。你的程序一共需要处理M (1 <= M < 50,000)个按输入次序到来的住店或退房的请求。第一个请求到来前，旅店中所有房间都是空闲的。
输入
* 第1行: 2个用空格隔开的整数：N、M
* 第2..M+1行: 第i+1描述了第i个请求，如果它是一个订房请求，则用2个数字 1、D_i描述，数字间用空格隔开；如果它是一个退房请求，用3 个以空格隔开的数字2、X_i、D_i描述
输出
第1..??行: 对于每个订房请求，输出1个独占1行的数字：如果请求能被满足 ，输出满足条件的最小的r；如果请求无法被满足，输出0

题目大意

Hotel有N（1 ≤ N ≤ 50,000）间rooms，并且所有的rooms都是连续排列在同一边，groups需要check in 房间，要求房间的编号为连续的r..r+Di-1并且r是最小的；visitors同样可能check out，并且他们每次check out都是编号为Xi ..Xi +Di-1 (1 ≤ Xi ≤ N-Di+1)的房间，题目的输入有两种样式：
1 a : groups需要check in a间编号连续的房间
2 a b ： visitors check out 房间，其中房间编号是 a…a+b-1
要求对于每次request，输出为groups分配数目为a的房间中编号最小的房间编号

解题思路

情况一：
最长区间在只在左孩子或只在右孩子
myq.sum(node)=max{myq.sum(node<<1),myq.sum(node<<1|1)；
情况二：
最长区间一部分在左孩子，一部分在右孩子
myq.sum(node)=myq.rmax(node<<1)+myq.lmax(node<<1|1)};
我们还需要记录：
每个区间从左边第一个孩子的最长连续区间；
右边第一个孩子的最长连续区间；
Lazy变量：
0：该点区间完全被覆盖；
1：该点区间完全未被覆盖；
-1：该点区间不完全被覆盖；
Lazy变量的状态：
1. lmax维护左区间的最大长度
2. rmax维护右区间的最大长度
3. sum维护区间1…N中的最大连续区间长度
注意
在添加元素和更新元素时都要更新lazy标记

完整代码

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#define SIZE 500000
#define myq l,m,node<<1
#define wbh m+1,r,node<<1|1
using namespace std;
struct Node{
    int sum;
    int len;
    int lazy;
    int lmax;
    int rmax;
}xmg[SIZE<<2];
void MYQ(int node)
{
    if(xmg[node<<1].sum==xmg[node<<1].len) xmg[node].lmax=xmg[node<<1].len+xmg[node<<1|1].lmax;
        else xmg[node].lmax=xmg[node<<1].lmax;
    if(xmg[node<<1|1].sum==xmg[node<<1|1].len) xmg[node].rmax=xmg[node<<1|1].len+xmg[node<<1].rmax;
        else xmg[node].rmax=xmg[node<<1|1].rmax;
    xmg[node].sum=max(max(xmg[node<<1].sum,xmg[node<<1|1].sum),xmg[node<<1].rmax+xmg[node<<1|1].lmax);
}
void MYQ_beautiful(int l,int r,int node)
{
    xmg[node].lazy=0;
    xmg[node].sum=xmg[node].len=xmg[node].lmax=xmg[node].rmax=r-l+1;
    if(l==r) return;
    int m=(l+r)>>1;
    MYQ_beautiful(myq);
    MYQ_beautiful(wbh);
}
void MYQ_young(int node)
{
    if(xmg[node].lazy==0) return;
    if(xmg[node].lazy==1)
    {
        xmg[node<<1].lazy=xmg[node<<1|1].lazy=1;
        xmg[node<<1].sum=xmg[node<<1].lmax=xmg[node<<1].rmax=0;
        xmg[node<<1|1].sum=xmg[node<<1|1].lmax=xmg[node<<1|1].rmax=0;
    }
    if(xmg[node].lazy==2)
    {
        xmg[node<<1].lazy=xmg[node<<1|1].lazy=2;
        xmg[node<<1].sum=xmg[node<<1].lmax=xmg[node<<1].rmax=xmg[node<<1].len;
        xmg[node<<1|1].sum=xmg[node<<1|1].lmax=xmg[node<<1|1].rmax=xmg[node<<1|1].len;
    }
    xmg[node].lazy=0;
}
void MYQ_cute(int L,int R,int c,int l,int r,int node)
{
    MYQ_young(node);
    if(L<=l&&r<=R)
    {
        if(c==1) xmg[node].sum=xmg[node].lmax=xmg[node].rmax=0;
            else xmg[node].sum=xmg[node].lmax=xmg[node].rmax=xmg[node].len;
        xmg[node].lazy=c;
        return;
    }   
    int m=(l+r)>>1;
    if(L<=m) MYQ_cute(L,R,c,myq);
    if(R>m) MYQ_cute(L,R,c,wbh);
    MYQ(node);
}
int MYQ_smart(int lenth,int l,int r,int node)
{
    MYQ_young(node);
    if(l==r) return 1;
    int m=(l+r)>>1;
    if(xmg[node<<1].sum>=lenth) return MYQ_smart(lenth,myq);
    else if(xmg[node<<1].rmax+xmg[node<<1|1].lmax>=lenth) return m-xmg[node<<1].rmax+1;
    else return MYQ_smart(lenth,wbh);
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    MYQ_beautiful(1,n,1);
    for(int i=1;i<=m;i++)
    {
        int flag,x,y;
        scanf("%d",&flag);
        if(flag==1)
        {
            scanf("%d",&x);
            if(xmg[1].sum>=x)
            {
                int f=MYQ_smart(x,1,n,1);
                printf("%d\n",f);
                MYQ_cute(f,f+x-1,1,1,n,1);
            }
            else printf("0\n");
        }
        else
        {
            scanf("%d%d",&x,&y);
            MYQ_cute(x,x+y-1,2,1,n,1);   
        }
    }

    return 0;
}