Cod sursa(job #2399696)

Utilizator eduardandrei20Nechifor Eduard Andrei eduardandrei20 Data 7 aprilie 2019 21:30:31
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.19 kb
#include <bits/stdc++.h>
#define dim 50005
#define oo 100000000
std::ifstream in("dijkstra.in");
std::ofstream out("dijkstra.out");
using namespace std;
int D[dim];int n,m;
bool viz[dim];
vector< pair<int,int> >graf[dim];
struct cmpD{
     bool operator()(int x,int y){
     return D[x]>D[y];
     }
};
priority_queue<int, vector<int>,cmpD>q;
void citire()
{
    in >> n >> m ;
     for( int i = 1; i<=m ;++i)
     {
         int x,y,c;
         in >> x >> y >> c ;
         graf[x].push_back(make_pair(y,c));
     }
}

void diji(int nod)
{
    for(int i =1; i<=n ;++i)
        if(i!=nod)D[i]=oo;
    for(int i = 1; i <= n; i++) {
        q.push(i);
    }

    while(!q.empty()) {
        int x = q.top();
        q.pop();
        viz[x] = true;
        for (size_t i = 0; i < graf[x].size(); i++) {
            int vecin = graf[x][i].first;
            int cost = graf[x][i].second;
            if (D[vecin] > D[x] + cost) {
                D[vecin] = D[x] + cost;
            }
        }
    }
}
int main()
{
citire();
diji(1);
for(int i=2;i<=n;i++)
       {
           if(D[i]==oo) out<<"0"<<' ';
           else out<<D[i]<<' ';
       }

    return 0;
}