首页 > 编程语言 > 详细

bzoj千题计划128：bzoj4552: [Tjoi2016&Heoi2016]排序

时间：2017-11-30 23:29:39 阅读：236 评论：0 收藏：0 [点我收藏+]

标签：using 一个输入长度 algo sample status gre 难题

http://www.lydsy.com/JudgeOnline/problem.php?id=4552

二分答案

把>=mid 的数看做1，<mid 的数看做0

这样升序、降序排列相当于区间查询0,1 的个数，区间覆盖0,1

线段树即可完成

查询给定位置p

如果=1，说明p位置的数>=mid ，上调下界

如果=0，说明p位置的数<mid，下调上界

#include<cstdio>
#include<iostream>
#include<algorithm>

using namespace std;

#define N 100001

int n,m,p;

int a[N],MID;

int sum0[N<<2],sum1[N<<2],flag[N<<2];

int tot0,tot1;

struct node
{
    int ty,l,r;
}e[N];

void read(int &x)
{
    x=0; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*10+c-‘0‘; c=getchar(); }
}

void build(int k,int l,int r)
{
    sum0[k]=sum1[k]=0;
    flag[k]=-1;
    if(l==r)
    {
        if(a[l]>=MID) sum1[k]++;
        else sum0[k]++;
        return;
    }
    int mid=l+r>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    sum0[k]=sum0[k<<1]+sum0[k<<1|1];
    sum1[k]=sum1[k<<1]+sum1[k<<1|1];
}

void down(int k,int l,int mid,int r)
{
    if(!flag[k])
    {
        sum0[k<<1]=mid-l+1;
        sum1[k<<1]=0;
        sum0[k<<1|1]=r-mid;
        sum1[k<<1|1]=0;
    }
    else
    {
        sum1[k<<1]=mid-l+1;
        sum0[k<<1]=0;
        sum1[k<<1|1]=r-mid;
        sum0[k<<1|1]=0;
    }
    flag[k<<1]=flag[k<<1|1]=flag[k];
    flag[k]=-1;
}

void query(int k,int l,int r,int opl,int opr)
{
    if(l>=opl && r<=opr) 
    {
        tot0+=sum0[k];
        tot1+=sum1[k];
        return;
    }
    int mid=l+r>>1;
    if(flag[k]!=-1) down(k,l,mid,r);
    if(opl<=mid) query(k<<1,l,mid,opl,opr);
    if(opr>mid) query(k<<1|1,mid+1,r,opl,opr);
}

void change(int k,int l,int r,int opl,int opr,int ty)
{
    if(l>=opl && r<=opr)
    {
        if(!ty) 
        {
            sum0[k]=r-l+1;
            sum1[k]=0;
        }
        else 
        {
            sum0[k]=0;
            sum1[k]=r-l+1;
        }
        flag[k]=ty;
        return;
    }
    int mid=l+r>>1;
    if(flag[k]!=-1) down(k,l,mid,r);
    if(opl<=mid) change(k<<1,l,mid,opl,opr,ty);
    if(opr>mid) change(k<<1|1,mid+1,r,opl,opr,ty);
    sum0[k]=sum0[k<<1]+sum0[k<<1|1];
    sum1[k]=sum1[k<<1]+sum1[k<<1|1];
}

int ask(int k,int l,int r,int pos)
{
    if(l==r) return sum1[k];
    int mid=l+r>>1;
    if(flag[k]!=-1) down(k,l,mid,r);
    if(pos<=mid) return ask(k<<1,l,mid,pos);
    return ask(k<<1|1,mid+1,r,pos);
}

bool check(int mid)
{
    MID=mid;
    build(1,1,n);
    for(int i=1;i<=m;++i)
    {
        tot0=tot1=0;
        query(1,1,n,e[i].l,e[i].r);
        if(!e[i].ty)
        {
            if(tot0) change(1,1,n,e[i].l,e[i].l+tot0-1,0);
            if(tot1) change(1,1,n,e[i].r-tot1+1,e[i].r,1);
        }
        else
        {
            if(tot1) change(1,1,n,e[i].l,e[i].l+tot1-1,1);
            if(tot0) change(1,1,n,e[i].r-tot0+1,e[i].r,0);
        }
    }
    return ask(1,1,n,p);
}

int main()
{
    read(n); read(m);
    for(int i=1;i<=n;++i) read(a[i]);
    for(int i=1;i<=m;++i) read(e[i].ty),read(e[i].l),read(e[i].r);
    read(p);
    int l=1,r=n,mid,ans;
    while(l<=r)
    {
        mid=MID=l+r>>1;
        if(check(mid)) ans=mid,l=mid+1;
        else r=mid-1;
    }
    cout<<ans;
}