acwing 849 Dijkstra求最短路 I 模板

地址 https://www.acwing.com/problem/content/description/851/

给定一个n个点m条边的有向图，图中可能存在重边和自环，所有边权均为正值。

请你求出1号点到n号点的最短距离，如果无法从1号点走到n号点，则输出-1。

输入格式

第一行包含整数n和m。

接下来m行每行包含三个整数x，y，z，表示存在一条从点x到点y的有向边，边长为z。

输出格式

输出一个整数，表示1号点到n号点的最短距离。

如果路径不存在，则输出-1。

数据范围

1≤n≤5001≤n≤500,
1≤m≤1051≤m≤105,
图中涉及边长均不超过10000。

```

输入样例：

3 3
1 2 2
2 3 1
1 3 4

输出样例：

3

```

#include <iostream>
#include <vector>
#include <algorithm>
#include <memory.h>


using namespace std;

const int MAX_N = 1010;

int gra[MAX_N][MAX_N];
int st[MAX_N];
int dist[MAX_N];
int n, m;

int solve()
{
    memset(dist, 0x3f3f3f3f, sizeof(dist));
    dist[1] = 0;

    //循环n-1 轮即可
    for (int i = 0; i < n - 1; i++) {
        int nearestNode = -1;
        //找到距离最近的一个点
        for (int j = 1; j <= n; j++) {
            if (st[j] == 0 && (nearestNode == -1 || dist[nearestNode] > dist[j])) {
                nearestNode = j;
            }
        }

        //用该点离第一号点的距离去更新其他点
        for (int j = 1; j <= n; j++) {
            dist[j] = min(dist[j], dist[nearestNode] + gra[nearestNode][j]);
        }


        st[nearestNode] = 1;
    }

    //如果n号点的距离没有更新 那么它不可达
    if (dist[n] == 0x3f3f3f3f) return -1;
    //返回最后更新的1号到n号的距离
    return dist[n];
}

int main()
{
    cin >> n >> m;
    //所有图边的长度初始化为0x3f3f3f3f
    memset(gra, 0x3f, sizeof gra);
    for (int i = 0; i < m; i++) {
        int a, b, c;
        scanf("%d %d %d",&a,&b,&c);
        //防止重边
        gra[a][b] = min(gra[a][b], c);
    }

    printf("%d\n",solve() ); 

    return 0;
}