Pagini recente » Cod sursa (job #900826) | Cod sursa (job #614112) | Cod sursa (job #1757845) | Cod sursa (job #295903) | Cod sursa (job #3125915)
/*
* Lefter Sergiu - 04/05/2023
*/
#include <fstream>
#include <vector>
#include <climits>
using namespace std;
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
const int N = 5e4;
const int INF = INT_MAX;
struct succesor {
int vf, c;
};
vector <succesor> s[N+1];
bool selectat[N+1];
int n, nh, m, d[N+1], h[N+1], poz[N+1];
void schimb(int p1, int p2) {
swap(h[p1], h[p2]);
poz[h[p1]] = p1;
poz[h[p2]] = p2;
}
void urca(int p) {
while (p > 1 && d[h[p]] < d[h[p/2]]) {
schimb(p, p/2);
p /= 2;
}
}
void adauga(int x) {
h[++nh] = x; //nh = nr-ul de elemente din heap
poz[x] = nh;
urca(nh);
}
void coboara(int p) {
int fs = 2 * p, fd = 2 * p + 1, bun = p;
if (fs <= nh && d[h[fs]] < d[h[bun]]) { //verific daca fs e in heap
bun = fs;
}
if (fd <= nh && d[h[fd]] < d[h[bun]]) { //verific daca fd e in heap
bun = fd;
}
if (bun != p) {
schimb(bun, p);
coboara(bun);
}
}
void sterge_1() { //sterge din varf
h[1] = h[nh--];
poz[h[1]] = 1;
coboara(1);
}
void dijkstra(int x0) {
for (int i = 1; i <= n; ++i) {
d[i] = INF;
selectat[i] = false;
}
d[x0] = 0;
adauga(x0); //adaug pe x0 in heap
while (nh > 0) { //cat timp heap-ul nu e vid
int x = h[1];
sterge_1(); //il stergem pe 1
selectat[x] = true;
for (auto p: s[x]) {
int y = p.vf;
int c = p.c;
if (selectat[y]) { //daca a fost selectat nu il mai bagam in seama
continue;
}
if (d[x] + c < d[y]) {
d[y] = d[x] + c;
//vedem daca e cazul doar sa urcam sau sa il adagaum in heap
if (poz[y] != 0) { //poz[y] = 0 pt cele care nu sunt in heap
urca(poz[y]); //i-am imbunatatit distanta, s-ar putea sa fie nevoie sa urce
} else {
adauga(y);
}
}
}
}
}
int main() {
in >> n >> m;
for (int i = 1; i <= m; ++i) {
int x, y, c;
in >> x >> y >> c;
s[x].push_back((succesor){y, c});
}
dijkstra(1);
for (int i = 2; i <= n; ++i) {
if (d[i] == INF) {
d[i] = 0;
}
out << d[i] << " ";
}
in.close();
out.close();
return 0;
}