跳跃游戏 (c++）

题目描述
这里有一个长度为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
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;
}