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;
}