Advertisement
apl-mhd

dijkstra using priorityQueue

May 5th, 2018
147
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.55 KB | None | 0 0
  1. #include <iostream>
  2. #include <cstdio>
  3. #include <stack>
  4. #include <queue>
  5. #include <algorithm>
  6. #include <vector>
  7. #define MAX_SZ 100
  8. #include <climits>
  9. #include <utility>
  10.  
  11.  
  12. using namespace std;
  13.  
  14. /*
  15. 5 10
  16. 0 1 10
  17. 0 3 5
  18. 1 2 1
  19. 1 3 2
  20. 2 4 4
  21. 3 1 3
  22. 3 2 9
  23. 3 4 2
  24. 4 0 7
  25. 4 2 6
  26. */
  27.  
  28. int n,m;
  29. vector< vector< pair<int, int> > >adj;
  30.  
  31.  
  32. //priority_queue< pair<int , int >,  vector< pair<int, int> >,  greater<pair<int ,int >>   > q;
  33. priority_queue<pair<int,int>, vector<pair<int,int>>, greater<pair<int,int>>>q;
  34.  
  35.  
  36.  
  37. void print_graph(){
  38.  
  39.     printf("n : %d, %d : m\n", n, m);
  40.  
  41.     for(int i=0; i<n; i++){
  42.  
  43.        // printf("%d -> ",i);
  44.  
  45.         for(int j=0; j < adj[i].size(); j++){
  46.  
  47.  
  48.  
  49.             printf("(%d, %d ) -> %d \n",i, adj[i][j].first, adj[i][j].second);
  50.         }
  51.  
  52.         printf("\n");
  53.     }
  54.  
  55. }
  56.  
  57. int dist[MAX_SZ];
  58. int visited[MAX_SZ];
  59. int parent[MAX_SZ];
  60.  
  61.  
  62. void init(int s){
  63.  
  64.     for(int i=0; i< n; i++){
  65.         dist[i] = INT_MAX;
  66.         visited[i] = 0;
  67.         parent[i] = -1;
  68.  
  69.     }
  70.  
  71.     dist[s]=0;
  72.  
  73. }
  74.  
  75. void relax(int u, int v, int w){
  76.  
  77.  
  78.  
  79. }
  80.  
  81. int min_dist_vertex(){
  82.  
  83.     return  0;
  84.  
  85. }
  86.  
  87.  
  88. void dijkstra(int s){
  89.  
  90.     init(s);
  91.     q.push(make_pair(0,0));
  92.  
  93.     while(!q.empty()){
  94.  
  95.         pair<int,int>p;
  96.  
  97.         p = q.top();
  98.         q.pop();
  99.  
  100.         int u = p.second;
  101.         int uwt = p.first;
  102.  
  103.         cout<<u<<" ";
  104.  
  105.         visited[u] = 1;
  106.  
  107.  
  108.         for (int i = 0; i < adj[u].size() ; ++i) {
  109.  
  110.             int v =adj[u][i].second;
  111.             int w = adj[u][i].first;
  112.  
  113.  
  114.             int tw = uwt + w;
  115.  
  116.             if(visited[v] == 0 && tw < dist[v]) {
  117.                 dist[v] = tw;
  118.                 q.push(make_pair(tw,v));
  119.  
  120.                 parent[v] = u;
  121.  
  122.             }
  123.  
  124.  
  125.         }
  126.  
  127.     }
  128.  
  129.  
  130. }
  131.  
  132. void printpath(int i){
  133.  
  134.     while( parent[i] !=-1){
  135.  
  136.  
  137.         int back = parent[i];
  138.  
  139.         cout<<back<<" ";
  140.  
  141.         i = parent[i];
  142.  
  143.     }
  144.  
  145.  
  146. }
  147.  
  148. void  printpathRecursion(int i){
  149.  
  150.  
  151.     if(parent[i] ==-1)
  152.         return;
  153.  
  154.         printpathRecursion(parent[i]);
  155.  
  156.     cout<<parent[i];
  157.  
  158.  
  159. }
  160.  
  161. int main()
  162. {
  163.     adj.resize(MAX_SZ);
  164.     freopen("graph2.txt", "r", stdin);
  165.     scanf("%d%d",&n,&m);
  166.  
  167.  
  168.  
  169.     for(int i=0; i<m; i++){
  170.         int u, v,w;
  171.  
  172.         scanf("%d%d%d",&u,&v,&w);
  173.         adj[u].push_back(make_pair(w,v));
  174.  
  175.     }
  176.  
  177.     //print_graph();
  178.  
  179.     dijkstra(0);
  180.  
  181.  
  182.     cout<<endl;
  183.  
  184.     for (int j = 0; j < n ; ++j) {
  185.  
  186.         cout<<j<<" "<<dist[j]<<endl;
  187.  
  188.     }
  189.  
  190.     printpath(2);
  191.  
  192.  //   printpathRecursion(2);
  193.  
  194.  
  195.  
  196.     return 0;
  197. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement