HDU 1540 POJ 2892 线段树区间合并

来源：互联网发布：美发店收银软件下载编辑：程序博客网时间：2024/06/07 01:12

给出N个点，M次操作，N个点开始在一条线上链式相连

D操作把某点删除掉

Q操作询问某点一共可以连接多少个点

R操作把上一次删除的点还原

线段树处理区间合并

分别记录每个区间的左端连续最长和右端连续最长

#include "stdio.h"#include "string.h"struct node{    int l,r,lx,rx,x;}data[200010];int mark[50100];int Max(int a,int b){    if (a<b) return b;else return a;}void build(int l,int r,int k){    int mid;    data[k].l=l;    data[k].r=r;    data[k].lx=data[k].rx=data[k].x=data[k].r-data[k].l+1;    if (l==r) return ;    mid=(l+r)/2;    build(l,mid,k*2);    build(mid+1,r,k*2+1);}void updata(int x,int k,int op){    int mid,ll,rr;    if (data[k].l==data[k].r)    {        data[k].lx=data[k].rx=data[k].x=op;        return ;    }    mid=(data[k].l+data[k].r)/2;    if (x<=mid) updata(x,k*2,op);    else        if (x>mid) updata(x,k*2+1,op);    ll=data[k*2].r-data[k*2].l+1;    rr=data[k*2+1].r-data[k*2+1].l+1;    data[k].lx=data[k*2].lx;    if (data[k*2].lx==ll) data[k].lx+=data[k*2+1].lx;    data[k].rx=data[k*2+1].rx;    if (data[k*2+1].rx==rr) data[k].rx+=data[k*2].rx;    data[k].x=Max(Max(data[k*2].x,data[k*2+1].x),data[k*2].rx+data[k*2+1].lx);}int query(int x,int k){    int mid;    if (data[k].l==data[k].r || data[k].x==0 || data[k].x==data[k].r-data[k].l+1)        return data[k].x;    mid=(data[k].l+data[k].r)/2;    if (x<=mid)    {        if (data[k*2].r-x+1<=data[k*2].rx)            return query(x,k*2)+query(mid+1,k*2+1);        else            return query(x,k*2);    }    else    {        if (x-data[k*2+1].l+1<=data[k*2+1].lx)            return query(x,k*2+1)+query(mid,k*2);        else            return query(x,k*2+1);    }}int main(){    int cnt,n,m,x;    char ch[10];    while(scanf("%d%d",&n,&m)!=EOF)    {        build(1,n,1);        cnt=0;        while (m--)        {            scanf("%s",ch);            if (ch[0]=='D')            {                scanf("%d",&x);                mark[cnt++]=x;                updata(x,1,0);            }            if (ch[0]=='Q')            {                scanf("%d",&x);                printf("%d\n",query(x,1));            }            if (ch[0]=='R')            {                cnt--;                if (cnt<0) cnt=0;                else updata(mark[cnt],1,1);            }        }    }    return 0;}

0 0