Advertisement
Mirbek

Дейкстра O(m * log(n))

Jan 9th, 2022
995
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.52 KB | None | 0 0
  1. /***
  2. Графы (Флойд, Дейкстра, Форд Белман)
  3. ***/
  4. #include <bits/stdc++.h>
  5.  
  6. using namespace std;
  7.  
  8. const int N = 5005;
  9. const int inf = 2e9;
  10.  
  11. int n, m;
  12. int dist[N];
  13. vector < pair <int, int> > g[N];
  14.  
  15. int main(){
  16.  
  17.     cin >> n >> m;
  18.  
  19.     for (int i = 1; i <= m; i++) {
  20.         int u, v, len;
  21.         cin >> u >> v >> len;
  22.         g[u].push_back(make_pair(v, len));
  23.         g[v].push_back(make_pair(u, len));
  24.     }
  25.  
  26.     for (int i = 0; i < N; i++) {
  27.         dist[i] = inf;
  28.     }
  29.  
  30.     int start;
  31.     cin >> start;
  32.     dist[start] = 0;
  33.  
  34.     set < pair <int, int> > st;
  35.     st.insert(make_pair(dist[start], start));
  36.  
  37.     while (!st.empty()) {
  38. //        int v = -1;
  39. //        for (int i = 1; i <= n; i++) {
  40. //            if (used[i] == 0 && (v == -1 || dist[v] > dist[i])) {
  41. //                v = i;
  42. //            }
  43. //        }
  44. //        if (v == -1) break;
  45.         int v = st.begin()->second;
  46.         st.erase(st.begin());
  47.  
  48.         for (int i = 0; i < g[v].size(); i++) {
  49.             int u = g[v][i].first, len = g[v][i].second;
  50.             if (dist[v] + len < dist[u]) {
  51.                 st.erase(make_pair(dist[u], u));
  52.                 dist[u] = dist[v] + len;
  53.                 st.insert(make_pair(dist[u], u));
  54.             }
  55.         }
  56.     }
  57.  
  58.     int finish;
  59.     cin >> finish;
  60.  
  61.     if (dist[finish] == inf) {
  62.         cout << "No path\n";
  63.     } else {
  64.         cout << dist[finish] << endl;
  65.     }
  66. }
  67. /**
  68. 6 8
  69. 1 2 2
  70. 1 3 3
  71. 3 5 1
  72. 2 5 2
  73. 2 4 4
  74. 5 4 1
  75. 5 6 5
  76. 4 6 2
  77. 1 6
  78. ***/
  79.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement