Cod sursa(job #3196229)

Utilizator suimerchantsui merchant suimerchant Data 23 ianuarie 2024 10:44:07
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.2 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
using namespace std;


ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");


struct muchie
{
    int vecin;
    int cost;
    bool operator < (const muchie &A) const
    {
        return cost > A.cost;
    }
};


int n,m;
int d[50005];
vector <muchie> g[50005];
priority_queue <muchie> q;


void read()
{
    fin>>n>>m;
    int a,b,c;
    for(int i=1;i<=m;i++)
    {
        fin>>a>>b>>c;
        g[a].push_back({b,c});
    }
    for(int i=2;i<=n;i++)d[i]=1e9;
}


void dijkstra(int nod)
{
    d[nod]=0;
    q.push({nod,0});
    while(!q.empty())
    {
        int nnod=q.top().vecin;
        q.pop();
        int s=g[nnod].size();
        for(int i=0;i<s;i++)
        {
            int vecin=g[nnod][i].vecin;
            int cost=g[nnod][i].cost;
            if(d[nnod]+cost<d[vecin])
            {
                d[vecin]=d[nnod]+cost;
                q.push({vecin,d[vecin]});
            }
        }
    }
}


int main()
{
    read();
    dijkstra(1);
    for(int i=2;i<=n;i++)
    {
        if(d[i]!=1e9)fout<<d[i]<<" ";
        else fout<<"0 ";
    }
    return 0;
}