Cod sursa(job #2425119)

Utilizator cochinteleandreeaCochintele Andreea Elena cochinteleandreea Data 24 mai 2019 12:50:30
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.15 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <utility>
#include <set>

using namespace std;
int n, i, j, c, m, k , pmax;
vector< pair<int,int > > a[50001];

set<pair<int,int> > of;

int main()
{
    ifstream f("dijkstra.in");
    ofstream g("dijkstra.out");

    f >> n >> m;

    int INF=(n-1)*20000;
    while(f >> i >> j >> c)
            a[i].push_back(make_pair(j,c));

    vector<int> v(n+1,0),d(n+1,INF);
    v[1] = 1;
    d[1] = 0;
    of.insert(make_pair(0,1));
    for(int i=1;i<=n;i++)
        {
            if(of.empty())break;
            int nod=(*of.begin()).second;
            of.erase(of.begin());

            for(auto i:  a[nod])
               {
                   int j=i.first;
                 int cost=i.second;


                   if(d[nod]+cost<d[j])
                   {
                       of.erase(make_pair(d[j],j));

                      d[j]=d[nod]+cost;
                      of.insert(make_pair(d[j],j));
                   }
               }

        }
    for(i = 2; i <= n; i ++)
        g << (d[i] != INF ? d[i] : 0) << " ";
    return 0;
}