[BZOJ4552]二分答案+线段树

说在前面

挺有意思的一道题，思想很奇特，一开始还不能理解。模拟几组数据之后粗略的有一点感觉

---

题目

BZOJ4552传送门

题意

给定一个N的排列，有M次操作。每次操作会将[L,R]区间内的数以升序或者降序排列，询问所有操作完成后，在p位置的数是多少。

输入与输出

第一行两个数字N，M。表示数列长度和局部排序的次数
接下来M行，每行三个数opt,L,R。表示对[L,R]区间进行一次排序，当opt为0时是降序，opt为1时是升序。
最后一行一个整数p，表示询问的位置

输出p位置上的数字

样例

sample input
6 3
1 6 2 5 3 4
0 1 4
1 3 6
0 2 4
3
sample output
5

---

解法

二分答案。
每次二分出一个mid，将线段树中所有小于等于mid的数变成0，大于mid的数变成1。每次区间排序的时候，相当于区间查询和区间修改。复杂度是O(Nlog²N)是可以过的。

这个题是具有二分性的，可以这么理解，每次排序相当于是把小于mid的丢到了左边，大于mid的丢到了右边。因为是排列，所以可以只用管大小关系。如果我们二分的mid偏小，那么1的个数就会增多，最后在p位置上的就是1。如果偏大或者刚好相等，p位置就会是0。不理解的可以手动模拟一下，可能会有助于理解。

---

自带大常数的代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;

int N , M , a[100005] , Qp ;
struct Node{
    int zer , one , flag ;
    Node *ls , *rs ;
    Node operator + ( const Node &A ) const {
        Node tmp ;
        tmp.zer = zer + A.zer ; tmp.one = one + A.one ;
        return tmp ;
    }
    void pushdown( const int &lf , const int &rg ){
        if( flag == -1 ) return ;
        int mid = ( lf + rg ) >> 1 ;
        ls->zer = ( mid - lf + 1 ) * (flag^1) ;
        ls->one = ( mid - lf + 1 ) * flag ;
        rs->zer = ( rg - mid ) * (flag^1) ;
        rs->one = ( rg - mid ) * flag ;
        ls->flag = rs->flag = flag ;
        flag = -1 ;
    }
    void updata(){
        zer = ls->zer + rs->zer ;
        one = ls->one + rs->one ;
    }
}w[20*100005] , *tw = w , *root ;
struct Ops{
    int opt , L , R ; 
}o[100005] ;

Node *build( int lf , int rg , const int &x ){
    Node *nd = ++tw ;
    if( lf == rg )
        nd->zer = ( a[lf] <= x ) , nd->one = ( a[lf] > x ) ;
    else{
        int mid = ( lf + rg ) >> 1 ;
        nd->ls = build( lf , mid , x ) ;
        nd->rs = build( mid+1 , rg , x ) ;
        nd->updata() ;
    }
    nd->flag = -1 ;
    return nd ;
}

void Modify( Node *nd , int lf , int rg , int L , int R , int x ){
    if( L <= lf && rg <= R ){
        //printf( "lf = %d  rg = %d\n" , lf , rg ) ;
        nd->zer = ( rg - lf + 1 ) * (1-x) ;
        nd->one = ( rg - lf + 1 ) * x ;
        nd->flag = x ;
    } else {
        nd->pushdown( lf , rg ) ;
        int mid = ( lf + rg ) >> 1 ;
        if( L <= mid ) Modify( nd->ls , lf  , mid , L , R , x ) ;
        if( R >  mid ) Modify( nd->rs , mid+1, rg , L , R , x ) ;
        nd->updata() ;
    }
}

Node Query( Node *nd , int lf , int rg , int L , int R ){
    if( L <= lf && rg <= R ){
        //printf( "lf = %d  rg = %d (%d %d)\n" , lf , rg , nd->zer , nd->one ) ;
        return *nd ; 
    }
    int mid = ( lf + rg ) >> 1 ;
    nd->pushdown( lf , rg ) ;
    if( R <= mid ) return Query( nd->ls , lf  , mid , L , R ) ;
    if( L >  mid ) return Query( nd->rs , mid+1, rg , L , R ) ;
    return Query( nd->ls , lf , mid , L , R ) + Query( nd->rs , mid+1, rg , L , R ) ;
}

void do_Options(){
    Node tmp ;
    for( int i = 1 ; i <= M ; i ++ ){
        int L = o[i].L , R = o[i].R , opt = o[i].opt ;
        tmp = Query( root , 1 , N , L , R ) ;
        //printf( "Q(%d %d) : zer %d  one %d\n" , L , R , tmp.zer , tmp.one ) ;
        if( !opt ){
            if( tmp.zer ) Modify( root , 1 , N , L , L + tmp.zer - 1 , 0 ) ;
            if( tmp.one ) Modify( root , 1 , N , R - tmp.one + 1 , R , 1 ) ;
        } else {
            if( tmp.one ) Modify( root , 1 , N , L , L + tmp.one - 1 , 1 ) ;
            if( tmp.zer ) Modify( root , 1 , N , R - tmp.zer + 1 , R , 0 ) ;
        }
    }
}

void solve(){
    int lf = 1 , rg = N , ans ;
    while( lf <= rg ){
        int mid = ( lf + rg ) >> 1 ;
        //printf( "MID  =  %d\n" , mid ) ;
        tw = w ; 
        root = build( 1 , N , mid ) ;
        do_Options() ;
        if( Query( root , 1 , N , Qp , Qp ).zer ){
            ans = mid ;
            rg = mid - 1 ;
        } else lf = mid + 1 ;
    }
    printf( "%d" , ans ) ;
}

int main(){
    scanf( "%d%d" , &N , &M ) ;
    for( int i = 1 ; i <= N ; i ++ ) scanf( "%d" , &a[i] ) ;
    for( int i = 1 ; i <= M ; i ++ )
        scanf( "%d%d%d" , &o[i].opt , &o[i].L , &o[i].R ) ;
    scanf( "%d" , &Qp ) ;
    solve() ;
}