【bzoj4592】【SHOI2015】【脑洞治疗仪】【线段树】

Description

曾经发明了自动刷题机的发明家SHTSC又公开了他的新发明:脑洞治疗仪--一种可以治疗他因为发明而日益增大的脑洞的神秘装置。
为了简单起见,我们将大脑视作一个01序列。1代表这个位置的脑组织正常工作,0代表这是一块脑洞。
1 0 1 0 0 0 1 1 1 0
脑洞治疗仪修补某一块脑洞的基本工作原理就是将另一块连续区域挖出,将其中正常工作的脑组织填补在这块脑洞中。
(所以脑洞治疗仪是脑洞的治疗仪?)
例如,用上面第8号位置到第10号位置去修补第1号位置到第4号位置的脑洞。我们就会得到:
1 1 1 1 0 0 1 0 0 0
如果再用第1号位置到第4号位置去修补第8号位置到第10号位置:
0 0 0 0 0 0 1 1 1 1
这是因为脑洞治疗仪会把多余出来的脑组织直接扔掉。
如果再用第7号位置到第10号位置去填补第1号位置到第6号位置:
1 1 1 1 0 0 0 0 0 0
这是因为如果新脑洞挖出来的脑组织不够多,脑洞治疗仪仅会尽量填补位置比较靠前的脑洞。
假定初始时SHTSC并没有脑洞,给出一些挖脑洞和脑洞治疗的操作序列,你需要即时回答SHTSC的问题:
在大脑某个区间中最大的连续脑洞区域有多大。

Input

第一行两个整数n,m。表示SHTSC的大脑可分为从1到n编号的n个连续区域。有m个操作。
以下m行每行是下列三种格式之一。
0 l r :SHTSC挖了一个从l到r的脑洞。
1 l0 r0 l1 r2 :SHTSC进行了一次脑洞治疗,用从l0到r0的脑组织修补l1到r1的脑洞。
2 l r :SHTSC询问l到r这段区间最大的脑洞有多大。
n,m <=200000,1<=l<=r<=n

Output

对于每个询问,输出一行一个整数,表示询问区间内最大连续脑洞区域有多大。

Sample Input

10 10
0 2 2
0 4 6
0 10 10
2 1 10
1 8 10 1 4
2 1 10
1 1 4 8 10
2 1 10
1 7 10 1 6
2 1 10

Sample Output

3
3
6

6

题解:

线段树维护区间1的个数,0的个数,区间长度,区间最长连续0,左起最长连续0,右起最长连续0.

这样可以解决0,2操作。

对于1操作先找到那个区间,然后在线段树上二分一下即可.

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 200010
using namespace std;
struct use{int s0,s1,sum,len,ls,rs;}t[N<<2];
int n,m,k,l,r,ll,rr,p[N<<2],num;
int read(){
  int x(0);
  char ch=getchar();
  while (ch<'0'||ch>'9') ch=getchar();
  while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
  return x; 
} 
use update(use a,use b){
  use c;
  c.s0=c.s1=c.sum=c.len=c.ls=c.rs=0;
  c.s0=a.s0+b.s0;c.s1=a.s1+b.s1;
  c.len=a.len+b.len;
  c.sum=max(max(a.sum,b.sum),a.rs+b.ls);
  if (a.s0==a.len) c.ls=a.len+b.ls;
  else c.ls=a.ls;
  if (b.s0==b.len) c.rs=b.len+a.rs;
  else c.rs=b.rs;
  return c;
}
void build(int k,int l,int r){
  int mid=(l+r)>>1;
  if (l==r){t[k].s1=t[k].len=1;return;}
  build(k<<1,l,mid);
  build(k<<1|1,mid+1,r);
  t[k]=update(t[k<<1],t[k<<1|1]);
}
void paint(int k,int l,int r,int v){
  if (v==0){
    t[k].ls=t[k].rs=t[k].s0=t[k].sum=t[k].len;
    t[k].s1=0;
  } 
  if(v==1){
    t[k].ls=t[k].rs=t[k].s0=t[k].sum=0;
    t[k].s1=t[k].len;
  } 
  p[k]=v;
}
void pushdown(int k,int l,int r){
  int mid=(l+r)>>1;
  paint(k<<1,l,mid,p[k]);
  paint(k<<1|1,mid+1,r,p[k]);
  p[k]=-1; 
}
void change(int k,int l,int r,int ll,int rr,int v){
  int mid=(l+r)>>1;
  if (ll<=l&&r<=rr){paint(k,l,r,v);return;}
  if (p[k]!=-1) pushdown(k,l,r);
  if (ll<=mid) change(k<<1,l,mid,ll,rr,v);
  if (mid<rr) change(k<<1|1,mid+1,r,ll,rr,v);
  t[k]=update(t[k<<1],t[k<<1|1]); 
} 
use query(int k,int l,int r,int ll,int rr){
   int mid=(l+r)>>1;
   if (ll==l&&r==rr) return t[k];
   if (p[k]!=-1) pushdown(k,l,r);
   if (rr<=mid) return query(k<<1,l,mid,ll,rr);
   else if (mid<ll) return query(k<<1|1,mid+1,r,ll,rr);
   else return update(query(k<<1,l,mid,ll,mid),query(k<<1|1,mid+1,r,mid+1,rr)); 
} 
void find(int k,int l,int r){
  int mid=(l+r)>>1;
  if (t[k].s0<=num){num-=t[k].s0;paint(k,l,r,1);return;}
  if (p[k]!=-1) pushdown(k,l,r);
  if (t[k<<1].s0>=num) find(k<<1,l,mid);
  else{
    num-=t[k<<1].s0;paint(k<<1,l,mid,1);
    find(k<<1|1,mid+1,r);
  }
  t[k]=update(t[k<<1],t[k<<1|1]);
}
void solve(int k,int l,int r,int ll,int rr){
  int mid=(l+r)>>1;
  if (ll<=l&&r<=rr){
     if (t[k].s0<=num) num-=t[k].s0,paint(k,l,r,1); 
     else find(k,l,r);
     return;
  }
  if (p[k]!=-1) pushdown(k,l,r);
  if (ll<=mid&&num) solve(k<<1,l,mid,ll,rr);
  if (rr>mid&&num) solve(k<<1|1,mid+1,r,ll,rr); 
  t[k]=update(t[k<<1],t[k<<1|1]);
}
int main(){
  n=read();m=read();
  for (int i=1;i<=(n<<2);i++) p[i]=-1;
  build(1,1,n);
  for (int i=1;i<=m;i++){
    k=read();
    if (k==0){
      l=read();r=read();
      change(1,1,n,l,r,0);
    }
    if (k==1){
      ll=read();rr=read();l=read();r=read();
      use t=query(1,1,n,ll,rr);
      change(1,1,n,ll,rr,0);
      num=t.s1;
      solve(1,1,n,l,r);
    }
    if (k==2){
      l=read();r=read();
      use t=query(1,1,n,l,r);
      printf("%d\n",t.sum);
    }
  } 
}


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sunshinezff/article/details/51468692
个人分类: 线段树
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭