Cod sursa(job #3003203)

Utilizator zoabaZob Alexandru Mihai zoaba Data 15 martie 2023 16:24:15
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.1 kb
#include <fstream>
#include <vector>
#include <queue>
#include <algorithm>

using namespace std;

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

typedef pair<int,int> ii;
typedef vector<int> vi;

const int NMAX = 50005;
const int INF = 1e9;
vector<ii>G[NMAX];
int n,m;

void dijkstra(int s)
{
    vi dist(n+2,INF);
    dist[s] = 0;
    priority_queue<ii, vector<ii>, greater<ii> > pq;
    pq.push(ii(0,s));
    while(!pq.empty())
    {
        ii fr = pq.top();
        pq.pop();
        int d = fr.first, u = fr.second;
        if(d > dist[u])
            continue;
        for(auto x : G[u])
        {
            if(dist[u] + x.second < dist[x.first])
            {
                dist[x.first] = dist[u] + x.second;
                pq.push(ii(dist[x.first],x.first));
            }
        }
    }
    for(int i=2;i<=n;++i)
        fout << dist[i] << ' ';
}

int main()
{
    int i,x,y,cost;
    fin >> n >> m;
    for(i=1;i<=m;++i)
    {
        fin >> x >> y >> cost;
        G[x].push_back(ii(y,cost));
    }

    dijkstra(1);
    return 0;
}