【TJOI & HEOI 2016】【JZOJ 4605】 【BZOJ 4552】排序

Description

序列长度为$n$，排序次数为$m$
$100\%1\leq n,m\leq 100000$
$Time\ Limits:6000ms$

Analysis

此题做法极为机智。
二分出一个数。把>=这个数的数变为1，<这个数的数变为0
按顺序做一遍区间排序，排序直接用线段树区间修改。
升序就0在前1在后，反之则反。
最后，看一下第q个位置上的数是0还是1，调整l,r继续二分。
$O((n+m)log_2n)$

Code

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=100010;
int n,m,n0,n1,a[N],c[N],sum[2][N];
struct segment
{
    int lz,n0,n1;
}tr[N*4];
struct Op
{
    int tp,l,r;
}b[N];
void down(int v,int l,int r)
{
    if(!tr[v].lz) return;
    tr[v+v].lz=tr[v+v+1].lz=tr[v].lz;
    int mid=(l+r)>>1;
    if(tr[v].lz==1)
    {
        tr[v+v].n0=mid-l+1,tr[v+v+1].n0=r-mid;
        tr[v+v].n1=tr[v+v+1].n1=0;
    }
    else
    {
        tr[v+v].n1=mid-l+1,tr[v+v+1].n1=r-mid;
        tr[v+v].n0=tr[v+v+1].n0=0;
    }
    tr[v].lz=0;
}
void change(int v,int l,int r,int x,int y,int z)
{
    if(l>r || x>y) return;
    if(l==x && r==y)
    {
        tr[v].lz=z+1;
        if(!z) tr[v].n0=r-l+1,tr[v].n1=0;
        else tr[v].n1=r-l+1,tr[v].n0=0;
        return;
    }
    down(v,l,r);
    int mid=(l+r)>>1;
    if(y<=mid) change(v+v,l,mid,x,y,z);
    else
    if(x>mid) change(v+v+1,mid+1,r,x,y,z);
    else
    change(v+v,l,mid,x,mid,z),change(v+v+1,mid+1,r,mid+1,y,z);
    tr[v].n0=tr[v+v].n0+tr[v+v+1].n0;
    tr[v].n1=tr[v+v].n1+tr[v+v+1].n1;
}
void query(int v,int l,int r,int x,int y)
{
    if(l==x && r==y)
    {
        n0+=tr[v].n0,n1+=tr[v].n1;
        return;
    }
    down(v,l,r);
    int mid=(l+r)>>1;
    if(y<=mid) query(v+v,l,mid,x,y);
    else
    if(x>mid) query(v+v+1,mid+1,r,x,y);
    else
    query(v+v,l,mid,x,mid),query(v+v+1,mid+1,r,mid+1,y);
}
int find(int v,int l,int r,int x)
{
    if(l==r) return tr[v].n0?0:1;
    down(v,l,r);
    int mid=(l+r)>>1;
    if(x<=mid) return find(v+v,l,mid,x);
    else return find(v+v+1,mid+1,r,x);
}
int main()
{
    scanf("%d %d",&n,&m);
    fo(i,1,n) scanf("%d",&a[i]);
    fo(i,1,m) scanf("%d %d %d",&b[i].tp,&b[i].l,&b[i].r);
    int q;
    scanf("%d",&q);
    int left=1,right=n,mid;
    while(left<right-1)
    {
        mid=(left+right)>>1;
        fo(i,1,n) change(1,1,n,i,i,a[i]>=mid);
        fo(i,1,m)
        {
            int l=b[i].l,r=b[i].r;
            n0=n1=0;
            query(1,1,n,l,r);
            if(b[i].tp) change(1,1,n,l,l+n1-1,1),change(1,1,n,l+n1,r,0);
            else change(1,1,n,l,l+n0-1,0),
            change(1,1,n,l+n0,r,1);
        }
        int x=find(1,1,n,q);
        if(x==0) right=mid;
        else left=mid;
    }
    printf("%d",left);
    return 0;
}