diff --git a/CSES Problem Set/Shortest Routes II.cpp b/CSES Problem Set/Shortest Routes II.cpp new file mode 100644 index 0000000..415f48e --- /dev/null +++ b/CSES Problem Set/Shortest Routes II.cpp @@ -0,0 +1,116 @@ +/* Problem URL: https://cses.fi/problemset/task/1672 */ + +#include + +using namespace std; + +#define V vector + +#define rmin(a, b) a = min(a, b) +#define rmax(a, b) a = max(a, b) + +#define rep(i, lim) for (size_t i = 0; i < (lim); i++) +#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++) + +#define repv(i, v) for (auto &i : (v)) +#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; } +#define sortv(v) sort(v.begin(), v.end()) +#define all(v) (v).begin(), (v).end() + +using vi = vector; +using vvi = vector; +using vvvi = vector; +using vvvvi = vector; + +using ll = long long; + +using vl = vector; +using vvl = vector; +using vvvl = vector; +using vvvvl = vector; + +template +auto operator<<(ostream &os, const vector &vec)->ostream& { + os << vec[0]; + for (size_t i = 1; i < vec.size(); i++) { + os << ' ' << vec[i]; + } + os << '\n'; + return os; +} + +template +auto operator>>(istream &is, vector &vec)->istream& { + for (auto &i : vec) { + is >> i; + } + return is; +} + +template +auto operator<<(ostream &os, const vector> &vec)->ostream& { + for (auto &i : vec) { + os << i[0]; + for (size_t j = 1; j < i.size(); j++) { + os << ' ' << i[j]; + } + os << '\n'; + } + return os; +} + +template +auto operator>>(istream &is, vector> &vec)->istream& { + for (auto &i : vec) { + for (auto &j : i) { + is >> j; + } + } + return is; +} + +int main() +{ + ios::sync_with_stdio(false); + cin.tie(nullptr); + + int n, m, q; + cin >> n >> m >> q; + + vvl weight(n, vl(n, INT64_MAX >> 1)); + vvl floyd(n, vl(n)); + + for (size_t i = 0; i < n; i++) { + weight[i][i] = 0; + } + + while (m--) { + ll a, b, c; + cin >> a >> b >> c; + a--, b--; + rmin(weight[a][b], c); + rmin(weight[b][a], c); + } + + for (size_t i = 0; i < n; i++) { + for (size_t j = 0; j < n; j++) { + floyd[i][j] = weight[i][j]; + } + } + + for (size_t i = 0; i < n; i++) { + for (size_t j = 0; j < n; j++) { + for (size_t k = 0; k < n; k++) { + rmin(floyd[j][k], floyd[j][i] + floyd[i][k]); + } + } + } + + while (q--) { + ll a, b; + cin >> a >> b; + a--, b--; + + cout << (floyd[a][b] != INT64_MAX >> 1 ? floyd[a][b] : -1) << '\n'; + } +}