首页 > 代码库 > hdu-1540-Tunnel Warfare-线段树

hdu-1540-Tunnel Warfare-线段树

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 55000
int 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;
}