hdu-1540-Tunnel Warfare-线段树

来源：互联网发布：淘宝是日本的吗编辑：程序博客网时间：2024/05/17 23:36

lmax[i]:i区间内，从左往右，连续的最大长度

rmax[i]:i区间内，从右往左，连续的最大长度

val[i]:i区间的长度。

#include<stdio.h>#include<string.h>#include<iostream>#include<algorithm>#include<math.h>#include<stack>using namespace std;#define INF 99999999#define lmin 1#define rmax n#define lson l,(l+r)/2,rt<<1#define rson (l+r)/2+1,r,rt<<1|1#define root lmin,rmax,1#define maxn 55000int lmax[maxn*4];int rmax[maxn*4];int val[maxn*4];void push_up(int rt){}void push_down(int rt){}void creat(int l,int r,int rt){    lmax[rt]=rmax[rt]=val[rt]=r-l+1;    if(l!=r)    {        creat(lson);        creat(rson);    }}void update(int x,int l,int r,int rt){    if(r<x||l>x)return ;    lmax[rt]=min(x-l,lmax[rt]);    rmax[rt]=min(rmax[rt],r-x);    if(l!=r)    {        update(x,lson);        update(x,rson);    }}void update1(int x,int l,int r,int rt){    if(r<x||l>x)return ;    if(l!=r)    {        update1(x,lson);        update1(x,rson);        lmax[rt]=lmax[rt<<1];        rmax[rt]=rmax[rt<<1|1];        if(lmax[rt<<1]==val[rt<<1])            lmax[rt]=lmax[rt<<1]+lmax[rt<<1|1];        if(rmax[rt<<1|1]==val[rt<<1|1])            rmax[rt]=rmax[rt<<1|1]+rmax[rt<<1];       // cout<<l<<" "<<r<<" "<<lmax[rt]<<" "<<rmax[rt]<<endl;    }    else lmax[rt]=rmax[rt]=1;}int query(int x,int l,int r,int rt){    if(l>x)    {        if(l-x>rmax[rt-1])return 0;        else return lmax[rt];    }    else if(r<x)    {        if(x-r>lmax[rt+1])return 0;        else return rmax[rt];    }    if(l+lmax[rt]-1>=x)return lmax[rt];    if(r-rmax[rt]+1<=x)return rmax[rt];    return query(x,lson)+query(x,rson);}int num[maxn];stack<int>st;int main(){    int n,m,x;    char str[101];    while(~scanf("%d%d",&n,&m))    {        creat(root);        memset(num,0,sizeof(num));        while(m--)        {            scanf("%s",str);            if(str[0]=='D')            {                scanf("%d",&x);                st.push(x);                num[x]++;                if(num[x]==1)                {                    update(x,root);                }            }            else if(str[0]=='Q')            {                scanf("%d",&x);                if(num[x]==0)cout<<query(x,root)<<endl;                else cout<<"0"<<endl;            }            else            {                x=st.top();                st.pop();                if(num[x]!=0)                {                    num[x]=0;                    update1(x,root);                }            }        }    }    return 0;}

0 0