POJ3667---线段树

最新推荐文章于 2019-08-20 11:32:04 发布

carter91620

最新推荐文章于 2019-08-20 11:32:04 发布

阅读量341

点赞数

分类专栏：数据结构文章标签： query build up

本文链接：https://blog.csdn.net/dicpu/article/details/7323152

版权

数据结构专栏收录该内容

10 篇文章 0 订阅

订阅专栏

//对连续空间的插入和删除
//snum---表示此节点下空间个数，ls---表示此节点从左开始的空间个数，rs---表示此节点从右开始空间个数
//在查询时，按左节点、左节点---右节点、右节点的次序查询
//在每次查询时都要向下更新；在查询成功时，要向上更新
#include <iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 50005
#define inf 123456789
typedef __int64 i64;
int n,m,t;
struct TNode
{
    int l,r,snum;
    int ls,rs;
} node[6*N];
void build(int root,int l,int r)
{
    int mid=(l+r)>>1,p=root<<1;
    node[root].l=l,node[root].r=r;
    node[root].snum=node[root].ls=node[root].rs=r-l+1;
    if(l==r) return ;
    build(p,l,mid);
    build(p+1,mid+1,r);
}
void update(int root)//向下更新
{
    int p=root<<1;
    if(node[root].snum==0)
    {
        node[root].ls=node[root].rs=0;
        node[p].snum=node[p+1].snum=0;
        node[p].ls=node[p].rs=node[p+1].ls=node[p+1].rs=0;
    }
    if(node[root].snum==node[root].r-node[root].l+1)
    {
        node[root].ls=node[root].rs=node[root].r-node[root].l+1;
        node[p].ls=node[p].rs=node[p].snum=node[p].r-node[p].l+1;
        node[p+1].ls=node[p+1].rs=node[p+1].snum=node[p+1].r-node[p+1].l+1;
    }
}
void up(int root)//向上更新
{
    int p=root<<1;
    node[root].snum=node[p].snum+node[p+1].snum;
    node[root].ls=node[p].ls;
    if(node[p].ls==node[p].r-node[p].l+1)
    {
        node[root].ls+=node[p+1].ls;
    }
    node[root].rs=node[p+1].rs;
    if(node[p+1].rs==node[p+1].r-node[p+1].l+1)
    {
        node[root].rs+=node[p].rs;
    }
}
void modin(int root,int l,int r)
{
    update(root);
    if(node[root].l==l&&node[root].r==r)
    {
        node[root].snum=node[root].ls=node[root].rs=0;
    }
    else
    {
        int mid=(node[root].l+node[root].r)>>1,p=root<<1;
        if(l<=mid) modin(p,l,min(r,mid));
        if(r>mid) modin(p+1,max(mid+1,l),r);
        up(root);
    }
}
void modout(int root,int l,int r)
{
    update(root);
    if(node[root].l==l&&node[root].r==r)
    {
        node[root].snum=node[root].ls=node[root].rs=r-l+1;
    }
    else
    {
        int mid=(node[root].l+node[root].r)>>1,p=root<<1;
        if(r<=mid) modout(p,l,r);
        else if(l>mid) modout(p+1,l,r);
        else
        {
            modout(p,l,mid);
            modout(p+1,mid+1,r);
        }
        up(root);
    }
}
bool query(int root,int len)
{
    update(root);
    if(len>node[root].snum) return 0;
    if(node[root].snum==node[root].r-node[root].l+1||node[root].ls>=len)
    {
        printf("%d\n",node[root].l);
        modin(root,node[root].l,node[root].l+len-1);
        up(root);
        return 1;
    }
    int p=root<<1;
    if(query(p,len))
    {
        up(root);
        return 1;
    }
    if(node[p].rs+node[p+1].ls>=len)
    {
        int prs=node[p].rs;
        modin(p,node[p].r-prs+1,node[p].r);
        modin(p+1,node[p+1].l,node[p+1].l+len-prs-1);
        up(root);
        printf("%d\n",node[p].r-prs+1);
        return 1;
    }
    if(query(p+1,len))
    {
        up(root);
        return 1;
    }
}
int main()
{
    int id,l,r;
    scanf("%d%d",&n,&m);
    build(1,1,n);
    while(m--)
    {
        scanf("%d",&id);
        if(id==1)
        {
            scanf("%d",&l);
            if(!query(1,l)) puts("0");
        }
        else
        {
            scanf("%d%d",&l,&r);
            modout(1,l,l+r-1);
        }
    }
    return 0;
}