[bzoj4552][TJOI&HEOI2016]排序

题目大意

有一个n的排列，进行m次操作，每次操作是将一个区间升序或降序排序。
请你输出m次操作后第p个位置的值。

二分答案

题解好机智！
我们二分答案x，然后就是判断a[p]>=x?
把原序列转化为01序列，0表示小于x，1表示大于等于x。
那么区间升序排序其实就是把0全放前面，1都放后面。
用线段树兹瓷区间赋值就好了。
然后只需要维护区间0的个数。

#include
#include
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=100000+10;
int a[maxn],sum[maxn*5],set[maxn*5],ask[maxn][3];
bool bz[maxn*5];
int i,j,k,l,r,mid,t,n,m,p;
void mark(int p,int l,int r,int v){
    sum[p]=(v?0:r-l+1);
    set[p]=v;
    bz[p]=1;
}
void down(int p,int l,int r){
    int mid=(l+r)/2;
    if (bz[p]){
        mark(p*2,l,mid,set[p]);
        mark(p*2+1,mid+1,r,set[p]);
        bz[p]=0;
    }
}
void change(int p,int l,int r,int a,int b,int v){
    if (a>b) return;
    if (l==a&&r==b){
        mark(p,l,r,v);
        return;
    }
    down(p,l,r);
    int mid=(l+r)/2;
    if (b<=mid) change(p*2,l,mid,a,b,v);
    else if (a>mid) change(p*2+1,mid+1,r,a,b,v);
    else change(p*2,l,mid,a,mid,v),change(p*2+1,mid+1,r,mid+1,b,v);
    sum[p]=sum[p*2]+sum[p*2+1];
}
int query(int p,int l,int r,int a,int b){
    if (l==a&&r==b) return sum[p];
    down(p,l,r);
    int mid=(l+r)/2;
    if (b<=mid) return query(p*2,l,mid,a,b);
    else if (a>mid) return query(p*2+1,mid+1,r,a,b);
    else return query(p*2,l,mid,a,mid)+query(p*2+1,mid+1,r,mid+1,b);
}
bool check(int x){
    fo(i,1,n) 
        if (a[i]<x) change(1,1,n,i,i,0);else change(1,1,n,i,i,1);
    fo(i,1,m){
        t=query(1,1,n,ask[i][1],ask[i][2]);
        if (ask[i][0]){
            change(1,1,n,ask[i][1],ask[i][2]-t,1);
            change(1,1,n,ask[i][2]-t+1,ask[i][2],0);
        }
        else{
            change(1,1,n,ask[i][1],ask[i][1]+t-1,0);
            change(1,1,n,ask[i][1]+t,ask[i][2],1);
        }
    }
    return !query(1,1,n,p,p);
}
int main(){
    scanf("%d%d",&n,&m);
    fo(i,1,n) scanf("%d",&a[i]);
    fo(i,1,m) scanf("%d%d%d",&ask[i][0],&ask[i][1],&ask[i][2]);
    scanf("%d",&p);
    l=1;r=n;
    while (l1)/2;
        if (check(mid)) l=mid;else r=mid-1;
    }
    printf("%d\n",l);
}