dijkstra算法模版,基本思路

首先需要注意的是dijkstra算法只能用来求正权无环图的最短路

模版一

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

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

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

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

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

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

数据范围
1≤n≤500,
1≤m≤105,
图中涉及边长均不超过10000。

输入样例:

3 3
1 2 2
2 3 1
1 3 4

输出样例:

3

稠密图使用朴素Dijkstra,dist表示起点到该点距离,st是维护一个(确没确定已经是最短距离的)集合,dijkstra重点在于去找一个离起点最近的点再用它向后更新,并且根据贪心可以确定它已经是最短距离的一部分了

#include 
#include 
#include 
using namespace std;

const int N = 510;
int g[N][N], dist[N], st[N], n, m;

int dijkstra()
{
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;
    for(int i=0;i dist[j]))
                t = j;
        }
        st[t] = true;
        // 更新距离
        for(int j=1;j<=n;j++)
        {
            dist[j] = min(dist[j], dist[t] + g[t][j]);
        }
    }
    if(dist[n] == 0x3f3f3f3f) return -1;
    else return dist[n];
}

int main()
{
    scanf("%d%d", &n, &m);
    memset(g, 0x3f, sizeof g);
    int a, b, c;
    while(m--)
    {
        scanf("%d%d%d", &a, &b, &c);
        // 防止重边
        g[a][b] = min(g[a][b], c);
    }
    printf("%d", dijkstra());
    return 0;
}
模版二

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

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

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

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

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

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

数据范围
1≤n,m≤1.5×105,
图中涉及边长均不小于 0,且不超过 10000。

输入样例:

3 3
1 2 2
2 3 1
1 3 4

输出样例:

3
稀疏图用堆优化的dijkstra,维护一个小根堆,注意first是distance,因为优先队列按第一个进行排序,堆优化的点在于优化了找离起点最近的点的这一步

#include 
#include 
#include 
#include 
using namespace std;

typedef pair<int, int> PII;
const int N = 1.5e5 + 10;
int h[N], e[N], ne[N], idx;
int n, m, w[N], dist[N];
bool st[N];
priority_queue<PII, vector<PII>, greater<PII>> q;

void add(int a, int b, int c)
{
    e[idx] = b;
    w[idx] = c;
    ne[idx] = h[a];
    h[a] = idx++;
}

int dijkstra()
{
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;
    q.push({0, 1});
    while(q.size())
    {
        PII t = q.top();
        q.pop();
        int ver = t.second, d = t.first;
        if(st[ver]) continue;
        st[ver] = true;
        for(int i=h[ver];i!=-1;i=ne[i])
        {
            int node = e[i];
            if(dist[node] > d + w[i])
            {
                dist[node] = d + w[i];
                q.push({dist[node], node});
            }
        }
    }
    if(dist[n] == 0x3f3f3f3f) return -1;
    else return dist[n];
}

int main()
{
    scanf("%d%d", &n, &m);
    int a, b, c;
    memset(h, -1, sizeof h);
    while(m--)
    {
        scanf("%d%d%d", &a, &b, &c);
        add(a, b, c);
    }
    printf("%d\n", dijkstra());
    return 0;
}

你可能感兴趣的:(c++,dijkstra)