acwing 850. Dijkstra求最短路 II 模板

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

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

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

输入格式
第一行包含整数n和m。

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

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

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

数据范围
1≤n,m≤105,
图中涉及边长均不小于0,且不超过10000
输入样例:
3 3
1 2 2
2 3 1
1 3 4
输出样例:
3

这里相比上一篇 Dijkstra模板 使用了堆记录点距离 从而节省了查找最短距离的开销

#include 
#include 
#include 
#include 

using namespace std;
typedef pair<int, int> PII;

const int N = 100010;
vectorint, int>>> gra;
int dist[N];
int st[N];
int n, m;

int solve()
{
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;
    //这里是距离在前 节点号在后
    priority_queue, greater> heap;
    heap.push({ 0, 1 });      // first存储距离,second存储节点编号

    while (heap.size()) {
        auto t = heap.top();
        heap.pop();

        int node = t.second; int distance = t.first;

        if (st[node]) continue;
        st[node] = true;
        //查看每个出边
        for (int i = 0; i < gra[node].size(); i++) {
            int newnode = gra[node][i].first;
            int len = gra[node][i].second;
            if (dist[newnode] > dist[node] + len) {
                dist[newnode] = dist[node] + len;
                heap.push({ dist[newnode],newnode });
            }
        }


    }

    if (dist[n] == 0x3f3f3f3f) return -1;
    return dist[n];
}



int main()
{
    //cin >> n >> m;
    scanf("%d %d",&n,&m);
    gra.resize(n + 1);

    for (int i = 0; i < m; i++) {
        int a, b, c;
        //cin >> a >> b >> c;
        scanf("%d %d %d",&a,&b,&c);
        //这里是 目的节点号在前 边长在后
        gra[a].push_back({ b,c });
    }

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

    return 0;
}

 

你可能感兴趣的:(acwing 850. Dijkstra求最短路 II 模板)