From 2a94def2d6a02fde95e918a85ad9a7d9f9fba0b1 Mon Sep 17 00:00:00 2001 From: Segcolt <9hmbzr275@mozmail.com> Date: Fri, 14 Feb 2025 16:36:32 -0300 Subject: [PATCH] Another problem. --- CSES Problem Set/Number Spiral.cpp | 111 +++++++++++++++++++++++++++++ 1 file changed, 111 insertions(+) create mode 100644 CSES Problem Set/Number Spiral.cpp diff --git a/CSES Problem Set/Number Spiral.cpp b/CSES Problem Set/Number Spiral.cpp new file mode 100644 index 0000000..8aac635 --- /dev/null +++ b/CSES Problem Set/Number Spiral.cpp @@ -0,0 +1,111 @@ +/* Problem URL: https://cses.fi/problemset/task/1071 */ + +#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 t; + cin >> t; + while (t--) { + ll x, y; + cin >> x >> y; + + if (x > y) { + ll sz = (x - 1) * (x - 1); + + ll pos1, pos2; + if (x & 1) { + pos1 = y; + pos2 = x; + } else { + pos1 = x; + pos2 = y; + } + + cout << pos1 + x - pos2 + sz << '\n'; + } else { + ll sz = (y - 1) * (y - 1); + + ll pos1, pos2; + if (y & 1) { + pos1 = y; + pos2 = x; + } else { + pos1 = x; + pos2 = y; + } + + cout << pos1 + y - pos2 + sz << '\n'; + } + } +}