BZOJ 4552 排序 TJOI2016&HEOI2016

Problem

BZOJ

Solution

据说暴力分挺多的??

由于是个排列,我们就可以二分答案,然后我们的问题就转化为了,判断询问的pos位与当前的mid值的大小关系。
把序列转化为01序列,那么我们在转移的时候,如果是升序排列,0就在左方,1在右方,降序排列是类似的操作,所以我们就只需要查询区间内1的个数,修改区间的数字即可。最后查询pos位是1还是0。
注意每次建树时要重置lazy标记。

Code

#include <algorithm>
#include <cstdio>
#define rg register
#define pushup(x) sum[x]=sum[x<<1]+sum[x<<1|1]
using namespace std;
const int maxn=100010;
struct data{int l,r,op;}q[maxn];
int n,m,pos,ans,a[maxn],sum[maxn<<2],lazy[maxn<<2];
template <typename Tp> inline void read(Tp &x)
{
    x=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
}
void input()
{
    read(n);read(m);
    for(rg int i=1;i<=n;i++) read(a[i]);
    for(rg int i=1;i<=m;i++) read(q[i].op),read(q[i].l),read(q[i].r);
    read(pos);
}
void pushdown(int l,int r,int m,int rt)
{
    sum[rt<<1]=(lazy[rt]?(m-l+1):0);sum[rt<<1|1]=(lazy[rt]?(r-m):0);
    lazy[rt<<1]=lazy[rt<<1|1]=lazy[rt];
    lazy[rt]=-1;
}
void build(int l,int r,int val,int rt)
{
    lazy[rt]=-1;
    if(l==r){sum[rt]=(a[l]>=val);return ;}
    int m=(l+r)>>1;
    build(l,m,val,rt<<1);
    build(m+1,r,val,rt<<1|1);
    pushup(rt);
}
void update(int l,int r,int L,int R,int val,int rt)
{
    if(L>R) return ;
    if(L<=l&&r<=R){lazy[rt]=val;sum[rt]=(val?(r-l+1):0);return ;}
    int m=(l+r)>>1;
    if(~lazy[rt]&&l!=r) pushdown(l,r,m,rt);
    if(L<=m) update(l,m,L,R,val,rt<<1);
    if(m<R) update(m+1,r,L,R,val,rt<<1|1);
    pushup(rt);
}
int query(int l,int r,int L,int R,int rt)
{
    if(L<=l&&r<=R) return sum[rt];
    int m=(l+r)>>1,res=0;
    if(~lazy[rt]&&l!=r) pushdown(l,r,m,rt);
    if(L<=m) res+=query(l,m,L,R,rt<<1);
    if(m<R) res+=query(m+1,r,L,R,rt<<1|1);
    return res;
}
int check(int k)
{
    int lc,rc;
    build(1,n,k,1);
    for(rg int i=1;i<=m;i++)
    {
        rc=query(1,n,q[i].l,q[i].r,1);
        lc=q[i].r-q[i].l+1-rc;
        if(q[i].op) swap(lc,rc);
        update(1,n,q[i].r-rc+1,q[i].r,q[i].op^1,1);
        update(1,n,q[i].l,q[i].l+lc-1,q[i].op,1);
    }
    return query(1,n,pos,pos,1);
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    input();
    int l=1,r=n,mid;
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(check(mid)) ans=mid,l=mid+1;
        else r=mid-1;
    }
    printf("%d\n",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/as_a_kid/article/details/79927569