码迷,mamicode.com
首页 > 其他好文 > 详细

(i) hdu1540 (区间合并)

时间:2019-07-23 00:37:18      阅读:147      评论:0      收藏:0      [点我收藏+]

标签:math   http   span   自己   printf   pos   define   情况   要求   

hdu 1540  Tunnel Warfare

如在阅读本文时遇到不懂的部分,请在评论区询问,或跳转 线段树总介绍

 

推荐先阅读 (线段树) (h) poj3667 (区间合并) 以获得线段树区间合并的一些知识

 

本题与上题(poj3667)基本相同,但要注意询问部分要求的是包含x的0区间长度

 

query (看看代码,倒也直观,很容易理解)

int query(int rt,int l,int r,int x){
    if(l==r||!MaxiLen[rt]||MaxiLen[rt]==r-l+1)return MaxiLen[rt];
    //找到x ||含x的区间全是0 || 含x的区间全是1
  //自己看看分别会返回什么...MaxiLen维护区间内最长的0
int mid=l+r>>1; if(x<=mid){ if(x>=mid-last[LS]+1)return last[LS]+pre[RS]; //夹中间 else return query(LS,l,mid,x); //LS }else{ if(x<=mid+pre[RS])return last[LS]+pre[RS]; //夹中间 else return query(RS,mid+1,r,x); //RS } }

 

左右去找x十分普通,重点是夹在中间的情况,要分左右区间的前驱后继来看;以及整个区间的特殊情况判定 if(l==r||!MaxiLen[rt]||MaxiLen[rt]==r-l+1) 这三种可以合并起来写

 

代码

/*i.hdu1540*/
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
using namespace std;
const int N=2e6+3;
int STK[N],HEAD=0;int n,m;
int pre[N<<2],last[N<<2],MaxiLen[N<<2];
#define LS (rt<<1)
#define RS (LS|1)

void build(int rt,int l,int r){
    pre[rt]=last[rt]=MaxiLen[rt]=r-l+1;
    if(l==r)return;
    int mid=l+r>>1;
    build(LS,l,mid);
    build(RS,mid+1,r);
}
void pushup(int rt,int l,int r){
    MaxiLen[rt]=max(max(MaxiLen[LS],MaxiLen[RS]),pre[RS]+last[LS]);
    int mid=l+r>>1;
    if(pre[LS]==mid-l+1)pre[rt]=mid-l+1+pre[RS];
    else pre[rt]=pre[LS];
    if(last[RS]==r-mid)last[rt]=r-mid+last[LS];
    else last[rt]=last[RS];
    return;
}
void update(int rt,int l,int r,int x,int p){
    if(l==r){pre[rt]=last[rt]=MaxiLen[rt]=p;return;}
    int mid=l+r>>1;
    if(x<=mid)update(LS,l,mid,x,p);
    else update(RS,mid+1,r,x,p);
    pushup(rt,l,r);
}
int query(int rt,int l,int r,int x){
    if(l==r||!MaxiLen[rt]||MaxiLen[rt]==r-l+1)return MaxiLen[rt];
    //找到x ||含x的区间全是0 || 含x的区间全是1
    int mid=l+r>>1;
    if(x<=mid){
        if(x>=mid-last[LS]+1)return last[LS]+pre[RS]; //夹中间
        else return query(LS,l,mid,x);                 //LS
    }else{
        if(x<=mid+pre[RS])return last[LS]+pre[RS];     //夹中间
        else return query(RS,mid+1,r,x);               //RS
    }
}
int main(){
    while(scanf("%d%d",&n,&m)!=EOF){
        build(1,1,n);
        char c=getchar();int x;
        while(m--){
            c=getchar();
            if(c==D){
                scanf("%d",&x);
                update(1,1,n,x,0);
                STK[++HEAD]=x;
            }
            else if(c==Q){
                scanf("%d",&x);
                printf("%d\n",query(1,1,n,x));
            }
            else update(1,1,n,STK[HEAD--],1);
            c=getchar();
        }
    }
    return 0;
}

 

(i) hdu1540 (区间合并)

标签:math   http   span   自己   printf   pos   define   情况   要求   

原文地址:https://www.cnblogs.com/lsy263/p/11229113.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!