P11796 【MX-X9-T0】『GROI-R3』NaiLong 题解

P11796 【MX-X9-T0】『GROI-R3』NaiLong 题解

题目传送门

形式化题意

• 对于每个项 $a_i(1<i<n)$，当 $a_{i-1}=0$ 或 $a_{i+1}=1$ 时成立。
• 对于 $a_1$ 和 $a_n$，当 $a_2=1$ 时前者成立，当 $a_{n-1}=0$ 后者成立。
• 按题意模拟即可。

代码实现

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n,k,a[1000];
int main (){
    bool f=0;
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    if(k==1){
        if(a[k+1]==1) cout<<"Yes";
        else cout<<"No";
    }
    else if(k==n){
        if(a[k-1]==0) cout<<"Yes";
        else cout<<"No";
    }
    else{
        if(a[k+1]==1||a[k-1]==0) cout<<"Yes";
        else cout<<"No";
    }
    return 0;
}