AT2165 Median Pyramid Hard

传送门

感觉性质挺好想的,就是二分答案怎么就是想不到呢
考虑先二分出一个值,比他大的设为1,比他小的设为0
然后就可以\(O(n)\)地推出第一行是0还是1:
1、如果没有两个相邻且相等的,那么中间那个也就是第一行的
2、如果有两个相邻且相等的,那么第一行显然就是离底层最近的那个相邻且相等的
性质就是如果有两个相邻且相等的,那么上一行这两个位置和当前是一样的
总时间复杂度\(O(nlong)\)
代码:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
void read(int &x){
    char ch;bool ok;
    for(ok=0,ch=getchar();!isdigit(ch);ch=getchar())if(ch=='-')ok=1;
    for(x=0;isdigit(ch);x=x*10+ch-'0',ch=getchar());if(ok)x=-x;
}
#define rg register
const int maxn=2e5+10;
int n,a[maxn],b[maxn];
bool check(int x){
    int sum=0;
    for(rg int i=1;i<=n;i++){
        b[i]=a[i]>=x;
        if(b[i])sum++;
        else sum--;
    }
    bool flag=0;
    for(rg int i=1;i<n;i++)if(b[i]==b[i+1])flag=1;
    if(!flag)return sum>0;
    int now=(n+1)>>1,now1=now;
    while(now1>1){
        if(b[now]==b[now+1])return b[now];
        if(b[now1]==b[now1-1])return b[now1];
        now++,now1--;
    }
}
int main(){
    read(n);n=(n<<1)-1;
    for(rg int i=1;i<=n;i++)read(a[i]);
    int l=1,r=n;
    while(l<=r){
        int mid=(l+r)>>1;
        if(check(mid))l=mid+1;
        else r=mid-1;
    }
    printf("%d\n",r);
}
posted @ 2019-04-14 10:43 蒟蒻--lichenxi 阅读(...) 评论(...) 编辑 收藏