c++ 图论 :dijkstra(堆优化版) 、Bellman_ford 算法

在图论中,Dijkstra 和 Bellman-Ford 算法是两种常用的单源最短路径算法。Dijkstra 算法适用于无负权边的图,而 Bellman-Ford 算法则可以处理负权边并检测负权回路。

一、Dijkstra 算法(堆优化版)

Dijkstra 算法使用优先队列(通常使用最小堆)来优化查找当前最小距离的顶点。堆优化版 Dijkstra 算法的时间复杂度为 O((V + E) log V)。

1. 算法思想
  1. 从源点开始,初始化源点到各顶点的距离,源点到自身的距离为 0,其他顶点距离为无穷大。
  2. 使用优先队列(最小堆)维护当前已知的最短路径顶点。
  3. 每次从堆中取出距离最小的顶点,更新其邻接顶点的距离。
  4. 重复上述步骤直到堆为空。
2. 代码实现

以下是堆优化版 Dijkstra 算法的 C++ 实现:

cpp 复制代码
#include <iostream>
#include <vector>
#include <queue>
#include <functional>
#include <climits>

using namespace std;

const int INF = INT_MAX;

struct Edge {
    int to;
    int weight;
};

void dijkstra(int start, vector<vector<Edge>>& graph, vector<int>& dist) {
    int n = graph.size();
    dist.assign(n, INF);
    dist[start] = 0;

    priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
    pq.push({0, start});

    while (!pq.empty()) {
        int d = pq.top().first;
        int u = pq.top().second;
        pq.pop();

        if (d > dist[u]) continue;

        for (auto& edge : graph[u]) {
            int v = edge.to;
            int weight = edge.weight;
            if (dist[u] + weight < dist[v]) {
                dist[v] = dist[u] + weight;
                pq.push({dist[v], v});
            }
        }
    }
}

int main() {
    int n, m;
    cout << "Enter number of vertices and edges: ";
    cin >> n >> m;
    vector<vector<Edge>> graph(n);

    cout << "Enter edges (u, v, w):" << endl;
    for (int i = 0; i < m; ++i) {
        int u, v, w;
        cin >> u >> v >> w;
        graph[u].push_back({v, w});
        graph[v].push_back({u, w}); // 如果是有向图,去掉这一行
    }

    int start;
    cout << "Enter start vertex: ";
    cin >> start;

    vector<int> dist;
    dijkstra(start, graph, dist);

    cout << "Shortest distances from vertex " << start << ":" << endl;
    for (int i = 0; i < n; ++i) {
        cout << "Vertex " << i << ": " << dist[i] << endl;
    }

    return 0;
}

二、Bellman-Ford 算法

Bellman-Ford 算法可以处理包含负权边的图,并且可以检测负权回路。其时间复杂度为 O(VE)。

1. 算法思想
  1. 初始化源点到各顶点的距离,源点到自身的距离为 0,其他顶点距离为无穷大。
  2. 对所有边进行 V-1 次松弛操作,即对每一条边 (u, v),如果 dist[u] + weight < dist[v],则更新 dist[v]。
  3. 进行第 V 次松弛操作,如果还能更新任何顶点的距离,则存在负权回路。
2. 代码实现

以下是 Bellman-Ford 算法的 C++ 实现:

cpp 复制代码
#include <iostream>
#include <vector>
#include <climits>

using namespace std;

const int INF = INT_MAX;

struct Edge {
    int from;
    int to;
    int weight;
};

bool bellman_ford(int start, vector<Edge>& edges, int n, vector<int>& dist) {
    dist.assign(n, INF);
    dist[start] = 0;

    for (int i = 0; i < n - 1; ++i) {
        for (auto& edge : edges) {
            if (dist[edge.from] != INF && dist[edge.from] + edge.weight < dist[edge.to]) {
                dist[edge.to] = dist[edge.from] + edge.weight;
            }
        }
    }

    for (auto& edge : edges) {
        if (dist[edge.from] != INF && dist[edge.from] + edge.weight < dist[edge.to]) {
            return false; // 存在负权回路
        }
    }

    return true;
}

int main() {
    int n, m;
    cout << "Enter number of vertices and edges: ";
    cin >> n >> m;
    vector<Edge> edges(m);

    cout << "Enter edges (u, v, w):" << endl;
    for (int i = 0; i < m; ++i) {
        cin >> edges[i].from >> edges[i].to >> edges[i].weight;
    }

    int start;
    cout << "Enter start vertex: ";
    cin >> start;

    vector<int> dist;
    if (bellman_ford(start, edges, n, dist)) {
        cout << "Shortest distances from vertex " << start << ":" << endl;
        for (int i = 0; i < n; ++i) {
            cout << "Vertex " << i << ": " << dist[i] << endl;
        }
    } else {
        cout << "Graph contains a negative-weight cycle." << endl;
    }

    return 0;
}

总结

  1. Dijkstra 算法(堆优化版)适用于无负权边的图,使用最小堆优化查找当前最小距离顶点。
  2. Bellman-Ford 算法适用于包含负权边的图,可以检测负权回路,但时间复杂度较高。

这两种算法各有优缺点,选择时需根据具体应用场景和图的特性进行权衡。

相关推荐
机器学习之心2 分钟前
一区北方苍鹰算法优化+创新改进Transformer!NGO-Transformer-LSTM多变量回归预测
算法·lstm·transformer·北方苍鹰算法优化·多变量回归预测·ngo-transformer
yyt_cdeyyds13 分钟前
FIFO和LRU算法实现操作系统中主存管理
算法
暮色_年华14 分钟前
Modern Effective C++item 9:优先考虑别名声明而非typedef
c++
重生之我是数学王子22 分钟前
QT基础 编码问题 定时器 事件 绘图事件 keyPressEvent QT5.12.3环境 C++实现
开发语言·c++·qt
alphaTao39 分钟前
LeetCode 每日一题 2024/11/18-2024/11/24
算法·leetcode
我们的五年1 小时前
【Linux课程学习】:进程程序替换,execl,execv,execlp,execvp,execve,execle,execvpe函数
linux·c++·学习
kitesxian1 小时前
Leetcode448. 找到所有数组中消失的数字(HOT100)+Leetcode139. 单词拆分(HOT100)
数据结构·算法·leetcode
做人不要太理性1 小时前
【C++】深入哈希表核心:从改造到封装,解锁 unordered_set 与 unordered_map 的终极奥义!
c++·哈希算法·散列表·unordered_map·unordered_set
程序员-King.1 小时前
2、桥接模式
c++·桥接模式
chnming19871 小时前
STL关联式容器之map
开发语言·c++