HDR003B. 「MCOI-06」Gerrymandering

https://hydro.ac/record/6156d32c2d8c78cdba6e733d
2024-11-27 17:56:28 +00:00 · 2021-10-01 18:16:54 +08:00 · 2021-10-01 18:16:54 +08:00 · d7e56856e0
commit d7e56856e0
parent 096f23c71e
1 changed files with 44 additions and 0 deletions
--- a/Hydro/HDR003B/HDR003B.cpp
+++ b/Hydro/HDR003B/HDR003B.cpp
@ -0,0 +1,44 @@
+#include <bits/stdc++.h>
+
+using namespace std;
+
+int t, n, m, k, c, l;
+vector<int> ans;
+
+int main() {
+    cin >> t;
+    while (t--) {
+        c = 1;
+        cin >> n >> m >> k;
+        if (n * m % k != 0) {
+            cout << "NO" << endl;
+        } else {
+            cout << "YES" << endl;
+            for (int i = 1; i <= n; i++) {
+                ans.clear();
+                if (i & 1) {
+                    for (int j = 1; j <= m; j++) {
+                        if (!--c) {
+                            l++;
+                            c = k;
+                        }
+                        ans.push_back(l);
+                    }
+                } else {
+                    for (int j = m; j >= 1; j--) {
+                        if (!--c) {
+                            l++;
+                            c = k;
+                        }
+                        ans.insert(ans.begin(), l);
+                    }
+                }
+                for (auto i : ans) {
+                    cout << i << ' ';
+                }
+                cout << endl;
+            }
+        }
+    }
+    return 0;
+}