Cod sursa(job #2116441)

Utilizator FunnyStockyMihnea Andreescu FunnyStocky Data 27 ianuarie 2018 17:15:11
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.14 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
const int nmax=50000,inf=1000000000;
struct graf
{
    int nod,cost;
};
struct data
{
    int nod;
};
vector<graf>v[nmax+5];
int best[nmax+5];
bool operator<(data a,data b)
{
    return best[a.nod]>best[b.nod];
}
priority_queue<data>q;
void DIJ(int nod)
{
    for(int i=0;i<v[nod].size();i++)
    {
        int nod_nou=v[nod][i].nod;
        if(best[nod_nou]==inf)
        {
            best[nod_nou]=best[nod]+v[nod][i].cost;
            q.push({nod_nou});
        }
    }
    q.pop();
    if(q.empty())
        return;
    int x=q.top().nod;
    DIJ(x);
}
int n,m;
int main()
{
    fin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int a,b,cost;
        fin>>a>>b>>cost;
        v[a].push_back({b,cost});
    }
    for(int i=1;i<=n;i++)
        best[i]=inf;
    best[1]=0;
    q.push({1});
    DIJ(1);
    for(int i=2;i<=n;i++)
    {
        if(best[i]==inf)
            fout<<"0 ";
        else
            fout<<best[i]<<" ";
    }
    return 0;
}
/**

**/