bzoj 4552 [Tjoi2016&Heoi2016]排序——二分答案

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

二分答案，把 >= mid 的设成1、< mid 的设成0，之后排序就变成区间赋值了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ls Ls[cr]
#define rs Rs[cr]
using namespace std;
const int N=1e5+,M=N<<;
int n,m,a[N],ps; bool b[N];
int tot,Ls[M],Rs[M],sm[M],len[M],tg[M];
struct Node{bool op;int l,r;}q[N];
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
void init(int l,int r,int cr)
{
  len[cr]=r-l+;
  if(l==r)return; int mid=l+r>>;
  ls=++tot;init(l,mid,ls);
  rs=++tot;init(mid+,r,rs);
}
void build(int l,int r,int cr)
{
  tg[cr]=-;//
  if(l==r){sm[cr]=b[l];return;}
  int mid=l+r>>;
  build(l,mid,ls); build(mid+,r,rs);
  sm[cr]=sm[ls]+sm[rs];
}
void pshd(int cr)
{
  if(tg[cr]<)return;tg[ls]=tg[rs]=tg[cr];
  sm[ls]=(tg[cr]?len[ls]:);sm[rs]=(tg[cr]?len[rs]:);
  tg[cr]=-;
}
void mdfy(int l,int r,int cr,int L,int R,bool k)
{
  if(L>R)return;////
  if(l>=L&&r<=R){tg[cr]=k;sm[cr]=(k?len[cr]:);return;}
  int mid=l+r>>;pshd(cr);
  if(L<=mid)mdfy(l,mid,ls,L,R,k);
  if(mid<R)mdfy(mid+,r,rs,L,R,k);
  sm[cr]=sm[ls]+sm[rs];
}
int query(int l,int r,int cr,int L,int R)
{
  if(l>=L&&r<=R)return sm[cr];
  int mid=l+r>>; pshd(cr);
  if(L>mid)return query(mid+,r,rs,L,R);
  if(R<=mid)return query(l,mid,ls,L,R);
  return query(l,mid,ls,L,R)+query(mid+,r,rs,L,R);
}
bool chk(int mid)
{
  for(int i=;i<=n;i++)b[i]=(a[i]>=mid);
  build(,n,);
  for(int i=;i<=m;i++)
    {
      int d=query(,n,,q[i].l,q[i].r);
      if(q[i].op)
    {
      mdfy(,n,,q[i].l,q[i].l+d-,);
      mdfy(,n,,q[i].l+d,q[i].r,);
    }
      else
    {
      mdfy(,n,,q[i].r-d+,q[i].r,);
      mdfy(,n,,q[i].l,q[i].r-d,);
    }
    }
  return query(,n,,ps,ps);
}
int main()
{
  n=rdn();m=rdn();tot=;init(,n,);
  for(int i=;i<=n;i++)a[i]=rdn();
  for(int i=;i<=m;i++)q[i].op=rdn(),q[i].l=rdn(),q[i].r=rdn();
  ps=rdn();
  int l=,r=n,ans;
  while(l<=r)
    {
      int mid=l+r>>;
      if(chk(mid))ans=mid,l=mid+;
      else r=mid-;
    }
  printf("%d\n",ans);
  return ;
}