Luogu P2824 [HEOI2016/TJOI2016]排序 线段树+脑子

只会两个$log$的$qwq$

---

我们二分答案：设答案为$ans$，则我们把$a[i]<=ans$全部设成$0$，把$a[i]>ans$全部设成$1$，扔到线段树里，这样区间排序（升序）就是求出$[l,r]$中$0$（或$1$）的个数$cnt$，然后对区间$[l,l+cnt-1]$赋值为$0$,对$[l+cnt,r]$赋值为$1$。最后查一下所求位置是$0$还是$1$来决定上下界改变方向。

#include<cstdio>
#include<iostream>
#define R register int
#define ls (tr<<1)
#define rs (tr<<1|1)
using namespace std;
inline int g() {
    R ret=,fix=; register char ch; while(!isdigit(ch=getchar())) fix=ch=='-'?-:fix;
    do ret=ret*+(ch^); while(isdigit(ch=getchar())); return ret*fix;
}
const int N=;
int tg[N<<],d[N<<],op[N],l[N],r[N];
int a[N],n,m,k;
inline void build(int tr,int l,int r,int vl) { tg[tr]=-;
    if(l==r) {d[tr]=(int)a[l]>vl; return;} R md=l+r>>;
    build(ls,l,md,vl),build(rs,md+,r,vl); d[tr]=d[ls]+d[rs];
}
inline void spread(int tr,int l,int r) { if(!~tg[tr]) return ; R md=l+r>>;
    tg[ls]=tg[rs]=tg[tr]; d[ls]=(md-l+)*tg[tr],d[rs]=(r-md)*tg[tr]; tg[tr]=-;
}
inline void change(int tr,int l,int r,int LL,int RR,int vl) {
    if(LL<=l&&r<=RR) {tg[tr]=vl,d[tr]=(r-l+)*vl; return ;} spread(tr,l,r); R md=l+r>>;
    if(LL<=md) change(ls,l,md,LL,RR,vl); if(RR>md) change(rs,md+,r,LL,RR,vl); d[tr]=d[ls]+d[rs];
}
inline int query(int tr,int l,int r,int LL,int RR) {
    if(LL<=l&&r<=RR) return d[tr]; spread(tr,l,r); R md=l+r>>,ret=;
    if(LL<=md) ret+=query(ls,l,md,LL,RR); if(RR>md) ret+=query(rs,md+,r,LL,RR); return ret;
}
inline int ck(int vl) {
    build(,,n,vl); for(R i=;i<=m;++i) {
        R t=query(,,n,l[i],r[i]); if(!t||t==r[i]-l[i]+) continue;
        if(!op[i]) change(,,n,l[i],r[i]-t,),change(,,n,r[i]-t+,r[i],);
        else change(,,n,l[i],l[i]+t-,),change(,,n,l[i]+t,r[i],);
    } return query(,,n,k,k);
}
signed main() {
#ifdef JACK
    freopen("NOIPAK++.in","r",stdin);
#endif
    n=g(),m=g(); for(R i=;i<=n;++i) a[i]=g();
    for(R i=;i<=m;++i) op[i]=g(),l[i]=g(),r[i]=g();
    k=g(); R LL=,RR=n; while(LL<RR) {
        R md=LL+RR>>; if(ck(md)) LL=md+; else RR=md;
    } printf("%d\n",LL);
}

---

2019.07.03