cp/cses/graph-algorithms/shortest-routes-i.cc
2025-08-31 19:16:59 -05:00

129 lines
2.2 KiB
C++

#include <bits/stdc++.h> // {{{
#include <version>
#ifdef __cpp_lib_ranges_enumerate
#include <ranges>
namespace rv = std::views;
namespace rs = std::ranges;
#endif
// https://codeforces.com/blog/entry/96344
#pragma GCC optimize("O2,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
using namespace std;
using i32 = int32_t;
using u32 = uint32_t;
using i64 = int64_t;
using u64 = uint64_t;
using f64 = double;
using f128 = long double;
#if __cplusplus >= 202002L
template <typename T>
constexpr T MIN = std::numeric_limits<T>::min();
template <typename T>
constexpr T MAX = std::numeric_limits<T>::max();
template <typename T>
[[nodiscard]] static T sc(auto&& x) {
return static_cast<T>(x);
}
template <typename T>
[[nodiscard]] static T sz(auto&& x) {
return static_cast<T>(x.size());
}
#endif
static void NO() {
std::cout << "NO\n";
}
static void YES() {
std::cout << "YES\n";
}
template <typename T>
using vec = std::vector<T>;
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define ff first
#define ss second
#ifdef LOCAL
#define db(...) std::print(__VA_ARGS__)
#define dbln(...) std::println(__VA_ARGS__)
#else
#define db(...)
#define dbln(...)
#endif
// }}}
void solve() {
u32 n, m;
cin >> n >> m;
vec<u64> delta(n + 1, MAX<u64>);
using Edge = pair<u64, u32>;
vec<vec<Edge>> graph(n + 1);
for (u32 i = 0; i < m; ++i) {
u32 u, v;
u64 w;
cin >> u >> v >> w;
graph[u].emplace_back(v, w);
}
priority_queue<Edge, vector<Edge>, greater<Edge>> pq;
pq.emplace(0, 1);
while (!pq.empty()) {
auto [w, u] = pq.top();
pq.pop();
if (w >= delta[u]) {
continue;
}
delta[u] = w;
for (auto& [v, W] : graph[u]) {
if (w + W < delta[v]) {
pq.emplace(w + W, v);
}
}
}
for (u32 i = 1; i <= n; ++i) {
cout << delta[i] << " \n"[i == n];
}
}
int main() { // {{{
std::cin.exceptions(std::cin.failbit);
#ifdef LOCAL
std::cerr.rdbuf(std::cout.rdbuf());
std::cout.setf(std::ios::unitbuf);
std::cerr.setf(std::ios::unitbuf);
#else
std::cin.tie(nullptr)->sync_with_stdio(false);
#endif
u32 tc = 1;
// std::cin >> tc;
for (u32 t = 0; t < tc; ++t) {
solve();
}
return 0;
}
// }}}