洛谷 P1135 奇怪的电梯__（刷题)bfs

从今天开始，我要刷一些bfs或dfs 的题目了，现在开始的每道题都不简单（对于我而言）加油吧！

今天课上20mins刷了一道题，bfs，随便挑的。

课上只得到90分；

回到家，冷静一下，就通过打表得到了100分！！

是个好的开始！

题目戳这里

代码见下：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6 + 5;

int n,a,b;
int x[maxn];
int used[maxn];
int k;
bool is=false;

struct node
{
    int step,floor;
};

void bfs()
{
    queue<node> q;
    q.push(node{0,a});
    while(!q.empty())
    {
        node p=q.front();
        q.pop();
        if(p.floor == b)
        {
            printf("%d\n",p.step);
            is=true;
            return;
        }
        k = p.floor+x[p.floor];
        if(k<=n && !used[k])
        {
            q.push(node{p.step+1,k});
            used[k]=1;
        }
        k = p.floor-x[p.floor];
        if(k>1 && !used[k])
        {
            q.push(node{p.step+1,k});
            used[k]=1;
        }
    }
}

int main()
{
    scanf("%d%d%d",&n,&a,&b);
    for(int i=1;i<=n;i++) scanf("%d",&x[i]);
    if(b == a+x[a] || b == a-a[x])
    {
        printf("1\n");
        return 0;
    }
    bfs();
    if(!is) printf("-1\n");
    return 0;
}