[bzoj4592] [Shoi2015]脑洞治疗仪

Posted 2020-07-26 czllgzmzl

　　题面无法直视系列。

　　中规中矩的线段树题。

　　涉及的操作有：区间赋值为0，计算区间内1的个数，区间赋值为1，求区间内最大的连续的1的个数。

  1 #include<cstdio>
  2 #include<iostream>
  3 #include<cstring>
  4 using namespace std;
  5 const int maxn=200233,mxnode=maxn<<1;
  6 int lc[mxnode],rc[mxnode],sz[mxnode],num0[mxnode],mxl0[mxnode],mxr0[mxnode],mx0[mxnode],tot;
  7 int tag[mxnode];
  8 int i,j,k,n,m,L,R,MXR,NUM,ans;
  9 bool first;
 10  
 11 int ra;char rx;
 12 inline int read(){
 13     rx=getchar(),ra=0;
 14     while(rx<\'0\'||rx>\'9\')rx=getchar();
 15     while(rx>=\'0\'&&rx<=\'9\')ra*=10,ra+=rx-48,rx=getchar();return ra;
 16 }
 17  
 18 inline int max(int a,int b){return a>b?a:b;}
 19 inline void upd(int x,int l,int r){
 20     num0[x]=num0[l]+num0[r],
 21     mxl0[x]=mxl0[l],mxr0[x]=mxr0[r];
 22     if(mxl0[l]==sz[l])mxl0[x]+=mxl0[r];
 23     if(mxr0[r]==sz[r])mxr0[x]+=mxr0[l];
 24     mx0[x]=max(max(mx0[l],mx0[r]),mxr0[l]+mxl0[r]);
 25 }
 26 inline void pushdown(int x){
 27     if(tag[x]==-1)return;
 28     int l=lc[x],r=rc[x];
 29     if(tag[x]==0)
 30         tag[l]=tag[r]=0,
 31         mx0[l]=mxl0[l]=mxr0[l]=num0[l]=sz[l],
 32         mx0[r]=mxl0[r]=mxr0[r]=num0[r]=sz[r];
 33     else
 34         tag[l]=tag[r]=1,
 35         mx0[l]=mxl0[l]=mxr0[l]=num0[l]=0,
 36         mx0[r]=mxl0[r]=mxr0[r]=num0[r]=0;
 37     tag[x]=-1;
 38 }
 39 void build(int a,int b){
 40     int x=++tot;
 41     sz[x]=b-a+1,tag[x]=-1;
 42     if(a==b)return;
 43     int mid=a+b>>1;
 44     lc[x]=tot+1,build(a,mid),rc[x]=tot+1,build(mid+1,b);
 45 }
 46 void cover0(int x,int a,int b){
 47     if(num0[x]==sz[x])return;
 48     if(L<=a&&R>=b){
 49         tag[x]=0,
 50         mx0[x]=mxl0[x]=mxr0[x]=num0[x]=sz[x];
 51         return;
 52     }
 53     pushdown(x);
 54     int mid=a+b>>1;
 55     if(L<=mid)cover0(lc[x],a,mid);
 56     if(R>mid)cover0(rc[x],mid+1,b);
 57     upd(x,lc[x],rc[x]);
 58 }
 59 void get1(int x,int a,int b){
 60     if(num0[x]==sz[x])return;
 61     if(L<=a&&R>=b){
 62         NUM+=sz[x]-num0[x],
 63         tag[x]=0,
 64         mx0[x]=mxl0[x]=mxr0[x]=num0[x]=sz[x];
 65         return;
 66     }
 67     pushdown(x);
 68     int mid=a+b>>1;
 69     if(L<=mid)get1(lc[x],a,mid);
 70     if(R>mid)get1(rc[x],mid+1,b);
 71     upd(x,lc[x],rc[x]);
 72 }
 73 void treat(int x,int a,int b){
 74     if(!num0[x]||!NUM)return;
 75     if(L<=a&&R>=b&&num0[x]<=NUM){
 76         NUM-=num0[x],tag[x]=1,
 77         mx0[x]=mxl0[x]=mxr0[x]=num0[x]=0;
 78         return;
 79     }
 80     pushdown(x);
 81     int mid=a+b>>1;
 82     if(L<=mid)treat(lc[x],a,mid);
 83     if(R>mid)treat(rc[x],mid+1,b);
 84     upd(x,lc[x],rc[x]);
 85 }
 86 void query(int x,int a,int b){
 87     if(L<=a&&R>=b){
 88         if(first)ans=mx0[x],MXR=mxr0[x],first=0;
 89         else{
 90             ans=max(ans,max(MXR+mxl0[x],mx0[x]));
 91             if(num0[x]==sz[x])MXR+=sz[x];else MXR=mxr0[x];
 92         }
 93         return;
 94     }
 95     pushdown(x);
 96     int mid=a+b>>1;
 97     if(L<=mid)query(lc[x],a,mid);
 98     if(R>mid)query(rc[x],mid+1,b);
 99 }
100 int main(){
101     n=read(),m=read();
102     build(1,n);int id,x,y,x1,y1;
103     while(m--){
104         id=read(),x=read(),y=read();
105         if(id==0)L=x,R=y,cover0(1,1,n);//,printf("      num0:%d   mxl0:%d  mxr0:%d\\n",num0[1],mxl0[1],mxr0[1]);
106         if(id==1){
107             x1=read(),y1=read(),NUM=0,L=x,R=y,get1(1,1,n);
108             L=x1,R=y1,treat(1,1,n);
109         }
110         if(id==2)
111             L=x,R=y,first=1,ans=0,query(1,1,n),printf("%d\\n",ans);
112     }
113     return 0;
114 }

View Code