Cod sursa(job #2723363)

Utilizator TudorChirila11Tudor Chirila TudorChirila11 Data 13 martie 2021 22:34:54
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1 kb
#include <bits/stdc++.h>
#define st first
#define nd second
#define pb push_back
#define PII pair<int,int>
#define N 50001
#define INF 20000*50000
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
int n, m, i, j, ans[N], x, y, c;
vector<PII> v[N];
bool viz[N];
priority_queue<PII,vector<PII>,greater<PII> > pq;
void dijk(int nod)
{
    for(i=1;i<=n;++i)
        if(i!=nod)
        ans[i]=INF;
    else ans[i]=0;
    pq.push({0,1});
    while(!pq.empty())
    {
        PII f=pq.top();
        viz[f.nd]=1;
        nod=f.nd;
        pq.pop();
        for(auto i:v[nod])
            if(!viz[i.st]&&ans[i.st]>ans[nod]+i.nd)
            {
                ans[i.st]=ans[nod]+i.nd;
                pq.push({i.nd,i.st});
            }
    }
}
int main()
{
    fin>>n>>m;
    for(i=1;i<=m;++i)
    {
        fin>>x>>y>>c;
        v[x].pb({y,c});
        v[y].pb({x,c});
    }
    dijk(1);
    for(i=2;i<=n;++i)
        fout<<ans[i]<<' ';
    return 0;
}