题目描述
这一天,他来到了一座深山的山脚下,因为只有这座深山中的一位隐者才知道这种药草的所在。但是上山的路错综复杂,由于小小猪的病情,晴天小猪想找一条需时最少的路到达山顶,但现在它一头雾水,所以向你求助。
山用一个三角形表示,从山顶依次向下有1段、2段、3段等山路,每一段用一个数字T(1< =T< =100)表示,代表晴天小猪在这一段山路上需要爬的时间,每一次它都可以朝左、右、左上、右上四个方向走(注意:在任意一层的第一段也可以走到本层的最后一段或上一层的最后一段)。
晴天小猪从山的左下角出发,目的地为山顶,即隐者的小屋。
输入数据
第一行有一个数$n(2<= n <= 1000)$,表示山的高度
从第二行至第$n+1$行,第$i+1$行有$i$个数,每个数表示晴天小猪在这一段山路上需要爬的时间。
输出数据
一个数,即晴天小猪所需要的最短时间。
样例
5
1
2 3
4 5 6
10 1 7 8
1 1 4 5 6
输出
10
算法1
(spfa)
时间复杂度 $O(nm)$
参考文献
https://www.cnblogs.com/candy99/p/5981533.html
C++ 代码
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
using namespace std;
typedef long long LL;
// spfa 算法
const int N = 1005 * 1005 / 2, INF = 0x3f3f3f3f;
int n; // 山的高度
int t[N]; // 每条路的长度
inline int id(int i, int j)
{
return i * (i - 1) / 2 + j;
}
int h[N], e[N], ne[N], idx;
inline void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
inline void ins(int a, int b) # 无向边
{
add(a, b);
add(b, a);
}
int d[N], st[N]; // d[N]是距离,st[N]是否在队列里
queue<int> q;
void spfa(int s)
{
memset(d, 0x3f, sizeof d);
d[s] = 0;
q.push(s);
while (q.size())
{
int u = q.front();
q.pop();
st[u] = 0;
for (int i = h[u]; ~i; i = ne[i])
{
int v = e[i],w = t[u];
if (d[v] > d[u] + w)
{
d[v] = d[u] + w;
if (!st[v])
{
q.push(v);
st[v] = 1;
}
}
}
}
}
int main()
{
memset(h, -1, sizeof h);
cin >> n;
for (int i = 1; i <= n; i++)
{
if (i != 1)
ins(id(i, 1), id(i, i));
if (i < n && i != 1)
{
add(id(i, 1), id(i + 1, i + 1));
add(id(i, i), id(i + 1, 1));
}
for (int j = 1; j <= i; j++)
{
cin >> t[id(i, j)];
if (j < i)
ins(id(i, j), id(i, j + 1));
if (i < n)
{
add(id(i, j), id(i + 1, j));
add(id(i, j), id(i + 1, j + 1));
}
}
}
spfa(id(1, 1));
printf("%d\n", d[id(n, 1)] + t[id(n, 1)]);
return 0;
}