aboutsummaryrefslogtreecommitdiff
path: root/store/works/life/2020-algorithm-contest/code/3.cpp
diff options
context:
space:
mode:
authorYuqian Yang <crupest@crupest.life>2025-02-28 23:13:39 +0800
committerYuqian Yang <crupest@crupest.life>2025-02-28 23:13:39 +0800
commitdc1f0c4c0096013799416664894c5194dc7e1f52 (patch)
tree2f5d235f778cd720f4c39ec3e56b77ba6d99f375 /store/works/life/2020-algorithm-contest/code/3.cpp
parent7299d424d90b1effb6db69e3476ddd5af72eeba4 (diff)
downloadcrupest-dc1f0c4c0096013799416664894c5194dc7e1f52.tar.gz
crupest-dc1f0c4c0096013799416664894c5194dc7e1f52.tar.bz2
crupest-dc1f0c4c0096013799416664894c5194dc7e1f52.zip
chore(store): move everything to store.
Diffstat (limited to 'store/works/life/2020-algorithm-contest/code/3.cpp')
-rw-r--r--store/works/life/2020-algorithm-contest/code/3.cpp38
1 files changed, 38 insertions, 0 deletions
diff --git a/store/works/life/2020-algorithm-contest/code/3.cpp b/store/works/life/2020-algorithm-contest/code/3.cpp
new file mode 100644
index 0000000..088cc5e
--- /dev/null
+++ b/store/works/life/2020-algorithm-contest/code/3.cpp
@@ -0,0 +1,38 @@
+#include <iostream>
+
+int main()
+{
+ int length, numRows;
+ std::cin >> length >> numRows;
+
+ if (numRows == 1)
+ {
+ for (int i = 1; i <= length; i++)
+ std::cout << i << ' ';
+ return 0;
+ }
+
+ const int count_per_group = numRows * 2 - 2;
+ for (int row = 0; row < numRows; row++)
+ {
+ if (row == 0)
+ {
+ for (int p = 0; p < length; p += count_per_group)
+ std::cout << p + 1 << ' ';
+ }
+ else if (row == numRows - 1)
+ {
+ for (int p = row; p < length; p += count_per_group)
+ std::cout << p + 1 << ' ';
+ }
+ else
+ {
+ bool former = true;
+ const auto former_gap = count_per_group - row * 2;
+ const auto latter_gap = count_per_group - former_gap;
+ for (int p = row; p < length; p += (former ? former_gap : latter_gap), former = !former)
+ std::cout << p + 1 << ' ';
+ }
+ }
+ return 0;
+}