Advertisement
iStrzalka

Dijkstra

Apr 21st, 2017
399
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.31 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. #include <set>
  4.  
  5. using namespace std;
  6.  
  7. struct node
  8. {
  9.     vector <int> polaczenia;
  10.     vector <int> waga;
  11.     int self = 10000000;
  12. };
  13.  
  14. node graph[1000 * 1000 + 10];
  15.  
  16. void DajStera(int START = 1)
  17. {
  18.     set < pair <int, int> >  verts;
  19.  
  20.     graph[START].self = 0;
  21.     verts.insert({0, START});
  22.  
  23.  
  24.     int akt,sasiad;
  25.     while(!verts.empty())
  26.     {
  27.         akt = verts.begin()->second;
  28.         verts.erase(verts.begin());
  29.  
  30.         for(int i = 0; i < graph[akt].waga.size(); i++)
  31.         {
  32.             sasiad = graph[akt].waga[i];
  33.  
  34.             if(graph[akt].self + graph[akt].waga[i] < graph[sasiad].self)
  35.             {
  36.                 if(verts.find({graph[sasiad].self, sasiad}) != verts.end())
  37.                     verts.erase({graph[sasiad].self, sasiad});
  38.  
  39.                 graph[sasiad].self = graph[akt].self + graph[akt].waga[i];
  40.  
  41.                 verts.insert({graph[sasiad].self, sasiad});
  42.             }
  43.         }
  44.     }
  45. }
  46.  
  47. int main()
  48. {
  49.     int n,m,from, to, cost;
  50.     cin >> n >> m;
  51.  
  52.     for(int i = 0;i < m; i++)
  53.     {
  54.         cin >> from >> to >> cost;
  55.  
  56.         graph[from].polaczenia.push_back(to);
  57.         graph[from].waga.push_back(cost);
  58.     }
  59.  
  60.     DajStera();
  61.  
  62.     for(int i = 0; i <= n; i++)
  63.         cout << graph[i].self << endl;
  64. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement