BZOJ4552 [Tjoi2016&Heoi2016]排序（洛谷P2824）

线段树 二分

BZOJ题目传送门
洛谷题目传送门

一道有思维难度的线段树所以我就去看题解了

先二分答案。设答案为x，那么把原序列转化一下， <x < x <script type="math/tex" id="MathJax-Element-6"> $\geq x$的为1。线段树维护区间内1的个数。那么排序的时候直接区间修改就行了，最后判断 $a[q]$是不是1就好了。

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
using namespace std;
struct tree{ int l,r,f,sum; }t[N<<2];
struct query{ int l,r,f; }q[N];
int n,m,k,a[N],b[N];
inline char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
inline int _read(){
    int x=0,f=1; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
void build(int x,int l,int r,int w){
    t[x]=(tree){l,r,-1,0}; int mid=l+r>>1;
    if (l==r) { t[x].sum=a[l]>=w; return; }
    build(x<<1,l,mid,w),build(x<<1|1,mid+1,r,w);
    t[x].sum=t[x<<1].sum+t[x<<1|1].sum;
}
inline void pshd(int x){
    t[x<<1].f=t[x<<1|1].f=t[x].f,t[x].f=-1;
    t[x<<1].sum=(t[x<<1].r-t[x<<1].l+1)*t[x<<1].f;
    t[x<<1|1].sum=(t[x<<1|1].r-t[x<<1|1].l+1)*t[x<<1|1].f;
}
void mdfy(int x,int l,int r,int w){
    if (t[x].l>r||t[x].r<l) return;
    if (t[x].l>=l&&t[x].r<=r){
        t[x].sum=(t[x].r-t[x].l+1)*w,t[x].f=w; return;
    }
    if (~t[x].f) pshd(x);
    mdfy(x<<1,l,r,w),mdfy(x<<1|1,l,r,w);
    t[x].sum=t[x<<1].sum+t[x<<1|1].sum;
}
int srch(int x,int l,int r){
    if (t[x].l>r||t[x].r<l) return 0;
    if (t[x].l>=l&&t[x].r<=r) return t[x].sum;
    if (~t[x].f) pshd(x);
    return srch(x<<1,l,r)+srch(x<<1|1,l,r);
}
inline bool pd(int x){
    build(1,1,n,x);
    for (int i=1,l,r;i<=m;i++){
        int sum=srch(1,l=q[i].l,r=q[i].r);
        if (q[i].f) mdfy(1,l,l+sum-1,1),mdfy(1,l+sum,r,0);
        else mdfy(1,l,r-sum,0),mdfy(1,r-sum+1,r,1);
    }
    return srch(1,k,k);
}
int main(){
    n=_read(),m=_read();
    for (int i=1;i<=n;i++) a[i]=_read();
    for (int i=1;i<=m;i++)
        q[i].f=_read(),q[i].l=_read(),q[i].r=_read();
    k=_read(); int l=1,r=n,mid,ans=0;
    while (l<=r)
        if (!pd(mid=l+r>>1)) r=mid-1;
        else ans=mid,l=mid+1;
    return printf("%d\n",ans),0;
}