洛谷 P1135 奇怪的电梯

bfs 奇怪的电梯

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
int n,a,b;
int g[205],f[205],flag;//输入的可执行楼层数组，答案数组
bool vis[205];//判断数组
int bfs()//广度优先搜索
{
    queue<int>q;//队列q
    q.push(a);//添加队尾元素a
    f[a] = 0;
    while(!q.empty())//若q不为空循环
    {
        int k = q.front();//k为q的队首元素，即当前楼层数
        if(k == b)//k与b相等即当前楼层为目标楼层时返回答案数组次数即答案
            return f[k];
        //vis[k] = true;
        //vis单独放在这标记是不对的,应该是入队就标记
        q.pop();//删除队首元素
        int x = k + g[k], y = k - g[k];//上到的层数，下到的层数
        if(x<=n&&!vis[x])//判断上楼是否符合标准
        {
            vis[x] = true;
            q.push(x);
            f[x] = f[k] + 1;

        }
        if(y>=1&&!vis[y])//判断下楼是否符合标准
        {
            vis[y] = true;
            q.push(y);
            f[y] = f[k] + 1;
        }

    }
    return -1;//都不可以返回-1
}
int main()
{
    cin >> n >> a >> b;
    memset(f,0,sizeof(f));
    memset(vis,false,sizeof(vis));
    for(int i = 1; i <= n; i++)
    {
        cin >> g[i];
    }
    cout << bfs() << endl;
    return 0;
}