题目描述
这里有一个长度为n非负整数数组 arr,你最开始位于该数组的起始下标 m 处。当你位于下标 i 处时,你可以跳到 i + arr[i] 或者 i - arr[i]。
请你判断自己是否能够跳到对应元素值为 0 的下标处。
注意,不管是什么情况下,你都无法跳到数组之外。
输入
测试样例由多组测试数据组成。每组测试数据第一行输入两个正整数 n (1 <= n <= 50000 ), m ( 0 <= m < n )
第二行输入n个正整数ni ( 0 <= ni <= 500 )
输出
若你可以跳到对应元素值为0的下标出,输出YES
否则输出NO
样例输入 Copy
7 5
4 2 3 0 3 1 2
5 2
3 0 2 1 2
样例输出 Copy
YES
NO
代码 (BFS)
#include<bits/stdc++.h>
using namespace std;
int vis[50005],s[50005];
int n,m,flag;
void bfs(int x){
queue<int>q;
q.push(x);
while(!q.empty()){
int len=q.size();
for(int i=0;i<len;i++){
int d=q.front();
q.pop();
if(d<0||d>=n){
continue;
}
if(vis[d]){
continue;
}
vis[d]=1;
if(s[d]==0){
flag=1;
cout<<"YES"<<endl;
return;
}
q.push(d+s[d]);
q.push(d-s[d]);
}
}
return;
}
int main(){
while(cin>>n>>m){
flag=0;
memset(vis,0,sizeof(vis));
for(int i=0;i<n;i++){
cin>>s[i];
}
bfs(m);
if(!flag) cout<<"NO"<<endl;
}
return 0;
}