Add CSES problem - Number Spiral

ithallotulio · Jul 24, 2024 · a0f6d59 · a0f6d59
1 parent 94b883c
commit a0f6d59
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/cses/introductory-problems/number-spiral.cpp b/cses/introductory-problems/number-spiral.cpp
@@ -0,0 +1,21 @@
+#include <bits/stdc++.h>
+
+using namespace std;
+
+int main() {
+    int t;
+    long long int y, x, mainDiagonal, mx, ans;
+    cin >> t;
+    while (t--) {
+        cin >> y >> x;
+        mx = max(y, x);
+        mainDiagonal = ((mx * mx) - mx + 1);
+        if (mx % 2 == 0) {
+            ans = mainDiagonal + y - x;
+        } else {
+            ans = mainDiagonal + x - y;
+        }
+        cout << ans << endl;
+    }
+    return 0;
+}