bzoj4552: [Tjoi2016&Heoi2016]排序（二分+线段树）

　　又是久违的1A哇...

　　好喵喵的题！二分a[p]，把大于mid的数改为1，小于等于mid的数改为0，变成01串后就可以用线段树进行那一连串排序了，排序后如果p的位置上的数为0，说明答案比mid小，如果为1，说明答案比mid大。

　　如何理解呢？我们的目的其实是让比a[p]大的数都为1，这样子p位置上刚好为0。如果p位置上为1，说明mid较小，a[p]>mid，把a[p]给标记成了1。如果p位置上为0，就是把a[p]<=mid，把a[p]标记成了0，但是这样还有一些大于a[p]的位置也是0，所以继续往小的地方逼近答案。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=,inf=1e9;
struct poi{int sum,tag;}tree[maxn];
int n,m,q;
int a[maxn],b[maxn],l[maxn],r[maxn],ty[maxn];
inline void read(int &k)
{
    int f=;k=;char c=getchar();
    while(c<''||c>'')c=='-'&&(f=-),c=getchar();
    while(c<=''&&c>='')k=k*+c-'',c=getchar();
    k*=f;
}
inline void pushup(int x){tree[x].sum=tree[x<<].sum+tree[x<<|].sum;}
inline void pushdown(int x,int l,int r)
{
    if(tree[x].tag==-)return;
    int mid=(l+r)>>;
    if(tree[x].tag==)
    {
        tree[x<<].sum=mid-l+;tree[x<<].tag=;
        tree[x<<|].sum=r-mid;tree[x<<|].tag=;
    }
    else tree[x<<].sum=tree[x<<|].sum=tree[x<<].tag=tree[x<<|].tag=;
    tree[x].tag=-;
}
void build(int x,int l,int r)
{
    tree[x].tag=-;
    if(l==r){tree[x].sum=b[l];return;}
    int mid=(l+r)>>;
    build(x<<,l,mid);build(x<<|,mid+,r);
    pushup(x);
}
void update(int x,int l,int r,int cl,int cr,int ty)
{
    if(cl<=l&&r<=cr){tree[x].sum=ty*(r-l+);tree[x].tag=ty;return;}
    pushdown(x,l,r);
    int mid=(l+r)>>;
    if(cl<=mid)update(x<<,l,mid,cl,cr,ty);
    if(cr>mid)update(x<<|,mid+,r,cl,cr,ty);
    pushup(x);
}
int query(int x,int l,int r,int cl,int cr)
{
    if(cl<=l&&r<=cr)return tree[x].sum;
    pushdown(x,l,r);
    int mid=(l+r)>>,ret=;
    if(cl<=mid)ret+=query(x<<,l,mid,cl,cr);
    if(cr>mid)ret+=query(x<<|,mid+,r,cl,cr);
    return ret;
}
bool check(int x)
{
    for(int i=;i<=n;i++)b[i]=a[i]>x;
    build(,,n);
    for(int i=;i<=m;i++)
    {
        int x=query(,,n,l[i],r[i]);
        if(ty[i])
        {
            if(x)update(,,n,l[i],l[i]+x-,);
            if(l[i]+x<=r[i])update(,,n,l[i]+x,r[i],);
        }
        else
        {
            if(l[i]<=r[i]-x)update(,,n,l[i],r[i]-x,);
            if(x)update(,,n,r[i]-x+,r[i],);
        }
    }
    return !query(,,n,q,q);
}
int main()
{
    read(n);read(m);
    for(int i=;i<=n;i++)read(a[i]);
    for(int i=;i<=m;i++)read(ty[i]),read(l[i]),read(r[i]);
    read(q);
    int l=,r=n;
    while(l<r)
    {
        int mid=(l+r)>>;
        if(check(mid))r=mid;
        else l=mid+;
    }
    printf("%d",l);
}