线段树处理 poj2892 hdu1540

线段树处理 poj2892 hdu1540

这道题想了很久 没有想出来思路 后面上网找了题解 利用区间左端点连续 和右端点连续 处理了区间的连续

sum数组存储连续村庄个数，lsum存储区间内从左起连续个数，rsum存储区间内从右起连续个数

#include <cstdio>
#include <cstring>
#include <stack>
#include <algorithm>
using namespace std ;
#define INF 0xfffffff
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn=55555;
int num[maxn<<1];
int sum[maxn<<2],lsum[maxn<<2],rsum[maxn<<2];
//sum数组存储连续村庄个数，lsum存储区间内从左起连续个数，rsum存储区间内从右起连续个数
void pushup(int rt,int m)
{
    lsum[rt]=lsum[rt<<1];
    rsum[rt]=rsum[rt<<1|1];
    if(lsum[rt]==(m-(m>>1))) lsum[rt]+=lsum[rt<<1|1];
    if(rsum[rt]==(m>>1)) rsum[rt]+=rsum[rt<<1];
    sum[rt]=max(max(sum[rt<<1],sum[rt<<1|1]),rsum[rt<<1]+lsum[rt<<1|1]);
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        lsum[rt]=rsum[rt]=sum[rt]=1;
        return ;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt,r-l+1);
}
void update(int key,int add,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=lsum[rt]=rsum[rt]=add?1:0;
        return ;
    }
    int m=(l+r)>>1;
    if(key<=m) update(key,add,lson);
    else update(key,add,rson);
    pushup(rt,r-l+1);
}
int query(int key,int l,int r,int rt)
{
    if(l==r)
    {
        return sum[rt];
    }
    int m=(l+r)>>1;
    int ans=0;
    int a=l+lsum[rt]-1;int b=r-rsum[rt]+1;
    if(key>=l&&key<=a) ans=lsum[rt];
    if(key>=b&&key<=r) ans=max(ans,rsum[rt]);
    int c=m-rsum[rt<<1]+1;int d=m+lsum[rt<<1|1];
    if(key>=c&&key<=d) ans=max(ans,rsum[rt<<1]+lsum[rt<<1|1]);
    if(ans) return ans;
    if(key<=m) return query(key,lson);
    else return query(key,rson);
}
int main()
{
    int n,m,a;
    char str[100];
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(1,n,1);
        int cnt=0;
        while(m--)
        {
            scanf("%s",str);
            if(str[0]=='D')
            {
                scanf("%d",&a);
                update(a,0,1,n,1);
                num[cnt++]=a;
            }
            else if(str[0]=='R')
            {
                int t=num[--cnt];
                update(t,1,1,n,1);
            }
            else
            {
                scanf("%d",&a);
                printf("%d\n",query(a,1,n,1));
            }
        }
    }
    return 0;
}