From dfbc0ab51d29d2d175e03134229669f8fc91280b Mon Sep 17 00:00:00 2001 From: crupest Date: Mon, 12 Oct 2020 21:01:59 +0800 Subject: import(life): Add 2020-algorithm-contest. --- works/life/2020-algorithm-contest/code/3.cpp | 38 ++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 works/life/2020-algorithm-contest/code/3.cpp (limited to 'works/life/2020-algorithm-contest/code/3.cpp') diff --git a/works/life/2020-algorithm-contest/code/3.cpp b/works/life/2020-algorithm-contest/code/3.cpp new file mode 100644 index 0000000..088cc5e --- /dev/null +++ b/works/life/2020-algorithm-contest/code/3.cpp @@ -0,0 +1,38 @@ +#include + +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; +} -- cgit v1.2.3