MX - X9A 题解

思路

开数组存每个人被指向的次数,根据次数是否大于零输出即可。

时间复杂度:

$O(n)$

空间复杂度:

$O(n)$

Code

#include <iostream>
using namespace std;
int main(){
    int n,k,p;
    cin>>n>>k;//n,k作用同题
    int a[n+1]={0};//存每人被指向次数
    for(int i=1;i<=n;i++){
        cin>>p;
        if(p==0){
            a[i+1]++;//指向下一个人
        }
        if(p==1){
            a[i-1]++;//指向上一个人
        }
    }
    if(a[k]){
        cout<<"Yes";
    }else{
        cout<<"No";
    }//进行判断输出
    return 0;//养成好习惯
}