bzoj 4592(洛谷 4344) [Shoi2015]脑洞治疗仪——线段树上二分

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=4592

1操作就是用线段树来二分找到第一个有 k 个０的位置。

在洛谷上A了，与暴力和网上题解对拍也都没问题。在bzoj上4msWA。不知道为什么。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ls Ls[cr]
#define rs Rs[cr]
using namespace std;
const int N=2e5+,M=N<<;
int n,m,sm[M],fl[M],fr[M],mx[M],Ls[M],Rs[M],lz[M],tot;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
  return fx?ret:-ret;
}
void build(int l,int r,int cr)
{
  sm[cr]=r-l+; lz[cr]=-; if(l==r)return;
  int mid=l+r>>;
  ls=++tot; build(l,mid,ls);
  rs=++tot; build(mid+,r,rs);
}
void pshd(int cr,int l,int mid,int r)
{
  if(lz[cr]==-)return;
  fl[ls]=fr[ls]=mx[ls]=(lz[cr]?:mid-l+); sm[ls]=(lz[cr]?mid-l+:);
  fl[rs]=fr[rs]=mx[rs]=(lz[cr]?:r-mid); sm[rs]=(lz[cr]?r-mid:);
  lz[ls]=lz[rs]=lz[cr]; lz[cr]=-;
}
void pshp(int cr)
{
  sm[cr]=sm[ls]+sm[rs];
  fl[cr]=fl[ls]+(sm[ls]?:fl[rs]);
  fr[cr]=fr[rs]+(sm[rs]?:fr[ls]);
  mx[cr]=max(max(fl[cr],fr[cr]),max(mx[ls],mx[rs]));
  mx[cr]=max(mx[cr],fr[ls]+fl[rs]);
}
void mdfy(int l,int r,int cr,int L,int R,bool k)
{
  if(l>=L&&r<=R)
    {
      fl[cr]=fr[cr]=mx[cr]=(k?:r-l+); sm[cr]=(k?r-l+:);
      lz[cr]=k; return;
    }
  int mid=l+r>>; pshd(cr,l,mid,r);
  if(L<=mid)mdfy(l,mid,ls,L,R,k);
  if(mid<R)mdfy(mid+,r,rs,L,R,k);
  pshp(cr);
}
int qry(int l,int r,int cr,int L,int R)
{
  if(l>=L&&r<=R) return sm[cr];
  int mid=l+r>>,ret=; pshd(cr,l,mid,r);
  if(L<=mid) ret=qry(l,mid,ls,L,R);
  if(mid<R) ret+=qry(mid+,r,rs,L,R);
  return ret;
}
int fnd(int l,int r,int cr,int L,int R,int k)
{
  if(l==r)return l;
  int mid=l+r>>; pshd(cr,l,mid,r);
  if(l>=L&&r<=R)
    {
      int siz=mid-l+-sm[ls];
      if(siz>=k)return fnd(l,mid,ls,L,R,k);
      return fnd(mid+,r,rs,L,R,k-siz);
    }
  if(mid<L)return fnd(mid+,r,rs,L,R,k);
  if(R<=mid)return fnd(l,mid,ls,L,R,k);
  int d=qry(l,mid,ls,L,R);
  d=mid-max(l,L)+-d;//L!!!!//max!!!!!probably just r>R
  if(d>=k) return fnd(l,mid,ls,L,R,k);
  return fnd(mid+,r,rs,L,R,k-d);
}
int query(int l,int r,int cr,int L,int R)
{
  if(l>=L&&r<=R)return mx[cr];
  int mid=l+r>>; pshd(cr,l,mid,r);
  if(mid<L) return query(mid+,r,rs,L,R);
  if(R<=mid) return query(l,mid,ls,L,R);
  int ret=max(query(l,mid,ls,L,R),query(mid+,r,rs,L,R));
  ret=max(ret,min(fr[ls],mid-L+)+min(fl[rs],R-mid));/////
}
void print()
{
  for(int i=;i<=n;i++)
    printf("%d ",qry(,n,,i,i));
  puts("");
}
int main()
{
  n=rdn(); m=rdn(); tot=; build(,n,);
  for(int i=,op,l,r,l1,r1,d,k;i<=m;i++)
    {
      op=rdn(); l=rdn(); r=rdn();
      if(!op)
    {
      mdfy(,n,,l,r,);
    }
      if(op==)
    {
      l1=rdn(); r1=rdn();
      d=qry(,n,,l,r); if(!d)continue;
      mdfy(,n,,l,r,);  k=qry(,n,,l1,r1);
      if(r1-l1+-k<=d) mdfy(,n,,l1,r1,);
      else
        {
          k=fnd(,n,,l1,r1,d);
          mdfy(,n,,l1,k,);
        }
    }
      if(op==)
    {
      printf("%d\n",query(,n,,l,r));
    }
    }
  return ;
}

附上暴力和maker和对拍，以便之后查。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2e5+;
int n,m,ans;
bool a[N];
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
  return fx?ret:-ret;
}
int main()
{
  n=rdn(); m=rdn(); for(int i=;i<=n;i++)a[i]=;
  for(int i=,op,l,r,l1,r1,d;i<=m;i++)
    {
      op=rdn(); l=rdn(); r=rdn();
      if(!op)
    {
      for(int j=l;j<=r;j++)a[j]=;
    }
      if(op==)
    {
      l1=rdn(); r1=rdn(); d=;
      for(int j=l;j<=r;j++)d+=a[j],a[j]=;
      for(int j=l1;j<=r1&&d;j++)
        d-=(!a[j]),a[j]=;
    }
      if(op==)
    {
      d=; ans=;
      for(int j=l;j<=r;j++)
          if(!a[j])d++;
          else ans=max(ans,d),d=;
      ans=max(ans,d);
      printf("%d\n",ans);
    }
    }
  return ;
}

暴力

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<ctime>
using namespace std;
const int N=2e5,M=;
int n,m,op,l,r;
int main()
{
  srand(time());
  n=rand()%N+; m=rand()%N+;
  printf("%d %d\n",n,m);
  for(int i=;i<=m;i++)
    {
      op=rand()%M;
      if(op>=)op=;
      l=rand()%n+; r=rand()%n+;
      if(l>r)swap(l,r);
      if(op==)
    {
      printf("%d %d %d ",op,l,r);
      l=rand()%n+; r=rand()%n+;
      if(l>r)swap(l,r);
      printf("%d %d\n",l,r);
    }
      else printf("%d %d %d\n",op,l,r);
    }
  return ;
}

maker

#include<cstdio>
#include<algorithm>
using namespace std;
int cnt;
int main()
{
  while()
    {
      system("./maker > data.in");
      system("./a < data.in > zj.out");
      system("./b < data.in > bl.out");
      if(system("diff zj.out bl.out"))return ;
      cnt++; printf("cnt=%d\n",cnt);
      if(cnt==)return ;
    }
}

对拍