Cod sursa(job #1833460)

Utilizator Cudrici_CarinaCudrici Carina Cudrici_Carina Data 22 decembrie 2016 12:43:38
Problema Algoritmul lui Dijkstra Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.17 kb
#include <fstream>
#include <vector>
#include <queue>

#define PII  pair<int, int>
#define VI   vector<int>

using namespace std;

ifstream fi("dijkstra.in");
ofstream fo("dijkstra.out");


const int Inf = 0x3f3f3f3f;
void ReadGraph();
void Dijkstra(int S, VI& d);

vector<PII> G[50001];
VI d;
int n, m;

int main()
{
    ReadGraph();
    Dijkstra(1, d);

    for (int i = 2; i <= n; ++i)
        if ( d[i] != Inf ) fo << d[i] << ' ';
                      else fo << 0 << ' ';


}

void ReadGraph()
{

    int x, y, w;
    fi >> n >> m;
    while (fi >> x >> y >> w)
        G[x].push_back({w,y});


}
void Dijkstra(int x, VI& d)
{
    d = VI(n + 1, Inf);
    d[x] = 0;
    priority_queue<PII, vector<PII>, greater<PII> > Q;

    for ( Q.push({0, x}); !Q.empty(); Q.pop() )
    {
        x = Q.top().second;
        int dx = Q.top().first;
        if ( dx > d[x] )  continue;
        for (size_t i=0;i<G[x].size();i++)
        {   PII p=G[x][i];
            int y = p.second, w = p.first;
            if ( d[y] > d[x] + w)
            {
                d[y] = d[x] + w;
                Q.push({d[y], y});
            }
        }
    }
}