JZOJ4605. 【TJOI2016&HEOI2016】排序

题目大意

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

Data Constraint
$n,m \leq 100000$

题解

先二分答案，不妨设为$mid$,问题转化为判断排完序后$a[q]$是否大于等于$mid$
把原序列转化为01序列，0表示小于x，1表示大于等于x。
那么区间升序排序其实就是把0全放前面，1都放后面。
然后用线段树维护一下区间内0的数量。

SRC

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

#define N 100000 + 10
const int MAXN = 17 ;
struct Operation {
    int op , l , r ;
} Mod[N] ;
struct Tree {
    int Len , Num0 , op ;
} T[MAXN*N] ;

int a[N] , tax[2] ;
int n , m , q ;
int st , ans , ret ;

Tree Merge( Tree a , Tree b ) {
    Tree ret ;
    ret.Len = a.Len + b.Len ;
    ret.Num0 = a.Num0 + b.Num0 ;
    ret.op = 0 ;
    return ret ;
}

void Download( int v ) {
    if ( !T[v].op ) return ;
    int l = v + v ;
    int r = v + v + 1 ;
    if ( T[v].op == 1 ) {
        if ( T[v].Num0 >= T[l].Len ) {
            T[l].Num0 = T[l].Len ;
            T[r].Num0 = T[v].Num0 - T[l].Len ;
        } else {
            T[l].Num0 = T[v].Num0 ;
            T[r].Num0 = 0 ;
        }
    } else {
        int Num1 = T[v].Len - T[v].Num0 ;
        if ( Num1 >= T[l].Len ) {
            T[l].Num0 = 0 ;
            T[r].Num0 = T[v].Num0 ;
        } else {
            T[l].Num0 = T[l].Len - Num1 ;
            T[r].Num0 = T[r].Len ;
        }
    }
    T[l].op = T[r].op = T[v].op ;
    T[v].op = 0 ;
}

void BuildTree( int v , int l , int r ) {
    if ( l == r ) {
        T[v].Len = 1 ;
        T[v].Num0 = ( a[l] < st ) ;
        T[v].op = 0 ;
        return ;
    }
    int mid = (l + r) / 2 ;
    BuildTree( v + v , l , mid ) ;
    BuildTree( v + v + 1 , mid + 1 , r ) ;
    T[v] = Merge( T[v+v] , T[v+v+1] ) ;
}

void Find( int v , int l , int r , int x , int y ) {
    if ( l == x && r == y ) {
        tax[0] += T[v].Num0 ;
        tax[1] += T[v].Len - T[v].Num0 ;
        return ;
    }
    Download( v ) ;
    int mid = (l + r) / 2 ;
    if ( y <= mid ) Find( v + v , l , mid , x , y ) ;
    else if ( x > mid ) Find( v + v + 1 , mid + 1 , r , x , y ) ;
    else {
        Find( v + v , l , mid , x , mid ) ;
        Find( v + v + 1 , mid + 1 , r , mid + 1 , y ) ;
    }
    T[v] = Merge( T[v+v] , T[v+v+1] ) ;
}

void Change( int v , int l , int r , int x , int y , int oper ) {
    if ( l == x && r == y ) {
        if ( oper == 1 ) {
            if ( tax[0] >= T[v].Len ) {
                T[v].Num0 = T[v].Len ;
                tax[0] -= T[v].Len ;
            } else {
                T[v].Num0 = tax[0] ;
                tax[1] -= T[v].Len - tax[0] ;
                tax[0] = 0 ;
            }
        } else {
            if ( tax[1] >= T[v].Len ) {
                T[v].Num0 = 0 ;
                tax[1] -= T[v].Len ;
            } else {
                T[v].Num0 = T[v].Len - tax[1] ;
                tax[0] -= T[v].Num0 ;
                tax[1] = 0 ;
            }
        }
        T[v].op = oper ;
        return ;
    }
    Download( v ) ;
    int mid = (l + r) / 2 ;
    if ( y <= mid ) Change( v + v , l , mid , x , y , oper ) ;
    else if ( x > mid ) Change( v + v + 1 , mid + 1 , r , x , y , oper ) ;
    else {
        Change( v + v , l , mid , x , mid , oper ) ;
        Change( v + v + 1 , mid + 1 , r , mid + 1 , y , oper ) ;
    }
    T[v] = Merge( T[v+v] , T[v+v+1] ) ;
}

void Search( int v , int l , int r , int x ) {
    if ( l == x && r == x ) {
        ret = !T[v].Num0 ;
        return ;
    }
    Download( v ) ;
    int mid = (l + r) / 2 ;
    if ( x <= mid ) Search( v + v , l , mid , x ) ;
    else Search( v + v + 1 , mid + 1 , r , x ) ;
    T[v] = Merge( T[v+v] , T[v+v+1] ) ;
}

bool Check() {
    tax[0] = tax[1] = 0 ;
    BuildTree( 1 , 1 , n ) ;
    for (int i = 1 ; i <= m ; i ++ ) {
        Find( 1 , 1 , n , Mod[i].l , Mod[i].r ) ;
        Change( 1 , 1 , n , Mod[i].l , Mod[i].r , Mod[i].op ) ;
    }
    ret = 0 ;
    Search( 1 , 1 , n , q ) ;
    return ret ;
}

void Solve() {
    ans = 1 ;
    int l = 1 , r = n ;
    while ( l <= r ) {
        st = (l + r) / 2 ;
        if ( Check() ) ans = st , l = st + 1 ;
        else r = st - 1 ;
    }
}

int main() {
    freopen( "sort.in" , "r" , stdin ) ;
    freopen( "sort.out" , "w" , stdout ) ;
    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" , &Mod[i].op , &Mod[i].l , &Mod[i].r ) , Mod[i].op ++ ;
    scanf( "%d" , &q ) ;
    Solve() ;
    printf( "%d\n" , ans ) ;
    return 0 ;
}

以上.