CF1349B Orac and Medians 题解

解题思路

一个比较显然的结论：

• 若序列中存在一段连续的长度大于 \(1\) 的区间，且区间中的数均为 \(k\)，那么一定可以将整个序列转化为 \(k\)。

进一步的，我们可以推出，若序列中有一段区间的中位数为 \(k\)，那么这个序列一定可以全部转化为 \(k\)。

最后特判 \(n=1\) 和序列中没有 \(k\) 的情况即可。

AC 代码

#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#define N 100005
#define inf 2e9+7
int n,k,a[N],p[N],ce;
int l[N],r[N],cl,cr;
inline void work(){
    scanf("%d%d",&n,&k);ce=0;
    for(register int i=1;i<=n;++i)
        scanf("%d",&a[i]);
    for(register int i=1;i<=n;++i)
        if(k==a[i]) p[++ce]=i;
    if(!ce){puts("No");return;}
    if(n==1){puts("Yes");return;}
    for(register int i=1;i<=ce;++i){
        int now=p[i];
        if(now>1&&a[now-1]>=k){
            puts("Yes");return;
        }
        else if(now<n&&a[now+1]>=k){
            puts("Yes");return;
        }
    }
    for(register int i=1;i<n;++i)
        if(a[i]>=k&&a[i+1]>=k){
            puts("Yes");return;
        }
    for(register int i=2;i<n;++i)
        if(a[i-1]>=k&&a[i+1]>=k){
            puts("Yes");return;
        }
    puts("No");
}
signed main(){
    int T;scanf("%d",&T);
    while(T--) work();
}