Cod sursa(job #2819218)

Utilizator alextmAlexandru Toma alextm Data 18 decembrie 2021 09:34:41
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.93 kb
#include <bits/stdc++.h>
using namespace std;

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

const int MAXN = 100001;
const int INF = (1 << 29);
typedef pair<int, int> PII;

int n, m, x, y, z, d[MAXN];
priority_queue <PII, vector<PII>, greater<PII>> q;
vector <PII> G[MAXN];

void Dijkstra(int start) {
  for(int i = 1; i <= n; i++)
    d[i] = INF;

  d[start] = 0;
  q.push({0, start});
  while(!q.empty()) {
    int dist = q.top().first;
    int node = q.top().second;
    q.pop();

    if(dist > d[node])
      continue;

    for(auto nxt : G[node])
      if(d[nxt.first] > dist + nxt.second) {
        d[nxt.first] = dist + nxt.second;
        q.push({d[nxt.first], nxt.first});
      }
  }
}

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

  Dijkstra(1);
  
  for(int i = 2; i <= n; i++)
		fout << (d[i] == INF ? 0 : d[i]) << " ";
	fout << "\n";

  return 0;
}