Cod sursa(job #2357435)

Utilizator Ciprian_PizzaVasile Capota Ciprian_Pizza Data 27 februarie 2019 13:24:05
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <bits/stdc++.h>
#define nmax 50002
#define oo 1e9

using namespace std;

int n,d[nmax];
vector < pair<int, int> > L[nmax];

void citire()
{
    ifstream fin("dijkstra.in");
    int x,y,c,m;
    fin >> n >> m;
    while(m--)
    {
        fin >> x >> y >> c;
        L[x].push_back({y,c});
        L[y].push_back({x,c});
    }
    fin.close();
}

void afisare(int k)
{
    ofstream fout("dijkstra.out");
    int i;
    for(i = 1; i <= n; i++)
        if(i != k) fout << d[i] << " ";
    fout.close();
}

void dijkstra(int k)
{
    int j,nod;
    for(j = 1; j <= n; j++)
        d[j] = oo;
    d[k] = 0;
    queue <int> q;
    q.push(k);
    while(!q.empty())
    {
        nod = q.front();
        q.pop();
        for(auto i : L[nod])
        {
            if(d[i.first] > d[nod] + i.second)
            {
                d[i.first] = d[nod] + i.second;
                q.push(i.first);
            }
        }
    }
    afisare(k);
}

int main()
{
    citire();
    dijkstra(1);
    return 0;
}