Codeforces GYM 101234A Hacker Cups and Balls

题目链接

题目大意

给定一个长为 n ( n 99999 ) 的序列,支持如下操作:

  • 选定一顿区间 [ l , r ] ,将 l r 的数按照升序(当 l < r )或降序(当 l > r

思路

考虑二分。二分中间这个数 x = m i d ,对于所有小于等于 m i d 的数都记为 1 ,大于都记为 0 ,完成所有排序操作后看正中间这个数是不是 1 ,若是 1 则说明答案至少大于等于 m i d ,将 l 右移。则问题就转化成了简单的区间覆盖和区间查询。祭上线段树,复杂度 O ( n log n )

坑点

区间覆盖的时候。由于 0 也是标记的一种,故在初始化的时候,要把标记数组赋成-1(此处应该WA8)

代码

#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<map>
#include<queue>
#include<vector>
#include<stack>
#include<set>
#include<cctype>
#define pa pair<int,int>
#define INF 0x3f3f3f3f
#define inf 0x3f
#define fi first
#define se second
#define mp make_pair
#define ll long long
#define ull unsigned long long
#define pb push_back

using namespace std;

inline ll read()
{
    long long f=1,sum=0;
    char c=getchar();
    while (!isdigit(c)) {if (c=='-') f=-1;c=getchar();}
    while (isdigit(c)) {sum=sum*10+c-'0';c=getchar();}
    return sum*f;
}
const int MAXN=100010;
struct Q{
    int l,r,opt;
}q[MAXN];
int f[4*MAXN],g[4*MAXN],a[MAXN],n,m;
void build(int root,int left,int right,int x)
{
//  f[root]=g[root]=0;
    if (left==right)
    {
        f[root]=(a[left]>=x);
        return ;
    }
    int mid=(left+right)>>1;
    build(2*root,left,mid,x);
    build(2*root+1,mid+1,right,x);
    f[root]=f[2*root]+f[2*root+1]; 
} 
void pushdown(int root,int x)
{
    g[2*root]=g[root],g[2*root+1]=g[root];
    f[2*root]=(x-x/2)*g[root],f[2*root+1]=g[root]*(x/2);
    g[root]=-1;
}
void update(int root,int left,int right,int qleft,int qright,int x)
{
    if (qleft<=left && right<=qright)
    {
        f[root]=(right-left+1)*x;
        g[root]=x;
        return ;
    }
    if (g[root]!=-1)
        pushdown(root,right-left+1);
    int mid=(left+right)>>1;
    if (qright<=mid)
        update(2*root,left,mid,qleft,qright,x);
    else if (qleft>mid)
        update(2*root+1,mid+1,right,qleft,qright,x);
    else
    {
        update(2*root,left,mid,qleft,mid,x);
        update(2*root+1,mid+1,right,mid+1,qright,x);
    }
    f[root]=f[2*root]+f[2*root+1];
}
int query(int root,int left,int right,int qleft,int qright)
{
    if (qleft<=left && right<=qright)
        return f[root];
    if (g[root]!=-1)
        pushdown(root,right-left+1);
    int mid=(left+right)>>1; 
    if (qright<=mid)
        return query(2*root,left,mid,qleft,qright);
    else if (qleft>mid)
        return query(2*root+1,mid+1,right,qleft,qright);
    else
    {
        int ans1,ans2;
        ans1=query(2*root,left,mid,qleft,mid);
        ans2=query(2*root+1,mid+1,right,mid+1,qright);
        return ans1+ans2;
    }
}
bool check(int mid)
{
    memset(f,0,sizeof(f));
    memset(g,-1,sizeof(g));
    build(1,1,n,mid);
    for (int i=1;i<=m;i++)
    {
        int tot=query(1,1,n,q[i].l,q[i].r);
        if (q[i].opt)
        {
            if (tot>0) update(1,1,n,q[i].l,q[i].l+tot-1,1);
            if (tot<q[i].r-q[i].l+1) update(1,1,n,q[i].l+tot,q[i].r,0);
        }
        else
        {
            if (tot>0) update(1,1,n,q[i].r-tot+1,q[i].r,1);
            if (tot<q[i].r-q[i].l+1) update(1,1,n,q[i].l,q[i].r-tot,0);
        }
    }
    return query(1,1,n,(n+1)/2,(n+1)/2);
}
int main()
{
    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",&q[i].l,&q[i].r);
        if (q[i].opt=(q[i].l>q[i].r))
            swap(q[i].l,q[i].r);
    }
    int l=1,r=n,mid,ans;
    while (l<=r)
    {
        mid=(l+r)>>1;
        if (check(mid))
            l=mid+1,ans=mid;
        else
            r=mid-1;
    }
    printf("%d",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/szh_0808/article/details/80925888