
今天来看看BFS, 首先,我们看题目,给你n个数字,你可以操作任意次数,让当前所在A楼层,到达目标楼层,求最小的操作数量,如果不能到达B层,就返回-1
这题可以用到许多图论算法,例如DFS,Floyd,BFS等,不过DFS被卡了,我们就简单看看BFS宽搜吧
首先,我们分析题目,我们可以把这个题目抽象成最短路问题,从A点到B点的最短距离,每次可以走
+k[i]步 或者 -k[i]步,求最短的耗费时间
下面详细的思考看BFS代码
#include <algorithm>
#include <iostream>
#include <cstring>
#include <queue>
struct PII {
int x, y;
};
using namespace std;
int n, a, b;
int f[210];
bool st[210];
int bfs(int x, int cnt)//当前楼层和计数器
{
queue<PII> q;//定义队列
q.push({x, cnt});//插入头节点
while (q.size()) {//等价于!empty()
PII t = q.front();//取出队头元素
q.pop();//删除队头元素
if (t.x == b) return t.y;//设置判断条件,当走到B点就return 时间(步数
if (t.x + f[t.x] <= n && !st[t.x + f[t.x]]) {//剪枝//电梯向上走
st[t.x + f[t.x]] = true;//将下一个节点设置为走过
q.push({t.x + f[t.x], t.y + 1});//插入下一个节点
}
if (t.x - f[t.x] >= 1 && !st[t.x - f[t.x]]) {//剪枝//电梯向下走
st[t.x - f[t.x]] = true;//将下一个节点设置为走过
q.push({t.x - f[t.x], t.y + 1});//插入下一个节点
}
}
return -1;//没找到就返回-1
}
signed main()
{
cin >> n >> a >> b;//输入楼层数,当前楼层位置, 目标楼层
for (int i = 1; i <= n; i++) {
cin >> f[i];//操作方案
}
cout << bfs(a, 0) << '\n';
return 0;
}
题目中N的数据范围最大为 200, 所以可以用下面这个
另外还有一种Floyd算法 O(n^3)
#include <algorithm>
#include <iostream>
#include <cstring>
using namespace std;
const int INF = 0x3f3f3f3f, N = 210;
int d[N][N];
void init()
{
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
if (i == j) d[i][j] = 0;
else d[i][j] = INF;
}
}
}
void floyd()
{
for (int k = 1; k <= n; k++) {
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
}
}
}
}
signed main()
{
cin >> n >> a >> b;
init();
for (int i = 1; i <= n; i++) {
int x; cin >> x;
if (i + x <= n) d[i][i + x] = 1;
if (i - x >= 1) d[i][i - x] = 1;
}
floyd();
if (d[a][b] != INF) cout << d[a][b];//d[a][b]为从a到b的最短路
else cout << "-1\n";
return 0;
}
好啦!今天的博客就结束啦!!
完结撒花~~~