[AGC06D] Median Pyramid Hard

AGC06D

题解：
由于求的是中位数，和数的大小关系有关，则可以二分答案，然后将所有大于等于当前二分的数的刷成$1$，小于的刷成$0$,然后根据$01$算出的答案调整二分区间。(套路)
然后问题就被简化成了给$2n-1$个数，每个数是0或1，求第一层的数。
令这$2n-1$个数是$b_1,b_2, \cdots ,b_{2n-1}$
随便找一个稍微长一点的01串就能发现：
1. 如果有连续的0或者1，令这个0,1区间是$[l,r]$$(r>l)$,则显然可以发现无论在那一层$[l,r]$的数不会变。 所以如果$b_n=b_{n+1}$或者$b_n=b_{n-1}$，则第一层的数就是$b_n$.
2. 如果有一段区间$[l,r](r>l)$满足 $b_i \neq b_{i+1} (l \leq i < r)$且($l=1$ 或 $b_l=b_{l-1}$)且 ($r=2n-1$或$b_r=b_{r+1}$) ,那么向上推一层的时候，整个$[l,r]$区间里的数会取反。这个区间对第一层有贡献当且今当$l=1,r=2n-1$. 证明是显然的，因为每向上推一层区间的长度都会减少$2$.
如果不满足$b_n=b_{n+1}$或者$b_n=b_{n-1}$， 则$b_n$一定存在于一个上述区间$[l_0,r_0]$,此时如果$n-l_0<r_0-n$则最终第一层的数就是$b_{l_0-1}$,否则就是$b_{r_0+1}$.(当$n-l_0=r_0-n$时一定有$b_{l_0-1}=b_{r_0+1}$) 特殊处理$l_0=1,r_0=2n-1$的情况即可.

时间复杂度$O(nlogn)$
写出来也就$30$行的样子。。

代码：

#include<bits/stdc++.h>
#define LL long long
#define ull unsigned long long
#define ULL ull
#define mp make_pair
#define pii pair<int,int>
#define piii pair<int, pii >
#define pll pair <ll,ll>
#define pb push_back
#define big 20160116
#define INF 2147483647
#define pq priority_queue
#define rank rk124232
#define y1 y20160116
#define y0 y20160110
using namespace std;
inline int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
namespace Mymath{
    LL qp(LL x,LL p,LL mod){
        LL ans=1;
        while (p){
            if (p&1) ans=ans*x%mod;
            x=x*x%mod;
            p>>=1;
        }
        return ans;
    }
    LL inv(LL x,LL mod){
        return qp(x,mod-2,mod);
    }
    LL C(LL N,LL K,LL fact[],LL mod){
        return fact[N]*inv(fact[K],mod)%mod*inv(fact[N-K],mod)%mod;
    }
    template <typename Tp> Tp gcd(Tp A,Tp B){
        if (B==0) return A;
        return gcd(B,A%B);
    }
    template <typename Tp> Tp lcm(Tp A,Tp B){
        return A*B/gcd(A,B);
    }
};
namespace fwt{
    using namespace Mymath;
    void FWT(int a[],int n,LL mod)
    {
        for(int d=1;d<n;d<<=1)
            for(int m=d<<1,i=0;i<n;i+=m)
                for(int j=0;j<d;j++)
                {
                    int x=a[i+j],y=a[i+j+d];
                    a[i+j]=(x+y)%mod,a[i+j+d]=(x-y+mod)%mod;
                    //xor:a[i+j]=x+y,a[i+j+d]=x-y;
                    //and:a[i+j]=x+y;
                    //or:a[i+j+d]=x+y;
                }
    }

    void UFWT(int a[],int n,LL mod)
    {
        LL rev=inv(2,mod);
        for(int d=1;d<n;d<<=1)
            for(int m=d<<1,i=0;i<n;i+=m)
                for(int j=0;j<d;j++)
                {
                    int x=a[i+j],y=a[i+j+d];
                    a[i+j]=1LL*(x+y)*rev%mod,a[i+j+d]=(1LL*(x-y)*rev%mod+mod)%mod;
                    //xor:a[i+j]=(x+y)/2,a[i+j+d]=(x-y)/2;
                    //and:a[i+j]=x-y;
                    //or:a[i+j+d]=y-x;
                }
    }
    void solve(int a[],int b[],int n,LL mod)
    {
        FWT(a,n,mod);
        FWT(b,n,mod);
        for(int i=0;i<n;i++) a[i]=1LL*a[i]*b[i]%mod;
        UFWT(a,n,mod);
    }
};
const int Maxn=2e5+5;
int a[Maxn];
int b[Maxn];
int n;
int check(int v){
    for (int i=1;i<=2*n-1;i++) b[i]=(a[i]>=v)?1:0;
    if (b[n]==b[n+1] || b[n-1]==b[n]){
        return b[n];
    }
    for (int l=n,r=n;l>=1;l--,r++){
        if (b[l]==b[l+1]){
            return b[l];
        }
        if (b[r]==b[r-1]){
            return b[r];
        }
    }
    return b[n]^(n&1)^1;
}
int main(){
    n=read();
    for (int i=1;i<=2*n-1;i++) a[i]=read();
    int lo=1,hi=2*n;
    while (hi-lo>1){
        int mid=lo+hi>>1;
        if (check(mid)){
            lo=mid;
        }
        else{
            hi=mid;
        }
    }
    printf("%d\n",lo);
    return 0;
}