POJ 3268 Silver Cow Party 最短路径 Dijkstra算法优化

最新推荐文章于 2024-10-16 19:08:56 发布

不吃不喝不玩就咔咔写代码

最新推荐文章于 2024-10-16 19:08:56 发布

阅读量48

点赞数

文章标签：算法数据结构图论

本文链接：https://blog.csdn.net/qq_53038151/article/details/132011147

版权

本文介绍了一个使用堆优化的Dijkstra算法来高效地寻找图中节点间的最短路径。相比于BellmanFord和Floyd算法，Dijkstra在处理某些情况时速度更快。代码示例展示了如何实现该算法并找到从特定起点到所有其他节点的最短距离。

摘要由CSDN通过智能技术生成

堆优化的Dijkstra算法确实非常快，求解最短路径比BellmanFord算法和Floyd算法快了太多。

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
typedef pair<int, int> P;
vector<P> edges[1007];
int d[1007], inf = 1000000000, N, M, X, fromEnd[1007];
bool used[1007];
void input()
{
    scanf("%d%d%d", &N, &M, &X);
    int from, to, cost;
    for (int i = 1; i <= M; i++)
    {
        scanf("%d%d%d", &from, &to, &cost);
        edges[from].push_back(P(cost, to));
    }
}
void dijkstra(int s)
{
    for (int i = 1; i <= N; i++)
    {
        d[i] = inf;
        used[i] = false;
    }
    d[s] = 0;
    priority_queue<P, vector<P>, greater<P>> que;
    que.push(P(0, s));
    while (!que.empty())
    {
        P current = que.top();
        que.pop();
        if (used[current.second] || current.first > d[current.second])
        {
            continue;
        }
        used[current.second] = true;
        for (int i = 0; i < edges[current.second].size(); i++)
        {
            P toEdge = edges[current.second][i];
            if (!used[toEdge.second] && d[current.second] + toEdge.first < d[toEdge.second])
            {
                d[toEdge.second] = d[current.second] + toEdge.first;
                que.push(P(d[toEdge.second], toEdge.second));
            }
        }
    }
}
void solve()
{
    dijkstra(X);
    for (int i = 1; i <= N; i++)
    {
        fromEnd[i] = d[i];
    }
    int ans = 0;
    for (int i = 1; i <= N; i++)
    {
        if (i == X)
        {
            continue;
        }
        dijkstra(i);
        if (fromEnd[i] + d[X] > ans)
        {
            ans = fromEnd[i] + d[X];
        }
    }
    printf("%d\n", ans);
}
int main()
{
    input();
    solve();
    return 0;
}