From a3a2519a087f710c8820d94584648aaba75528d7 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 7 May 2020 23:17:16 +0800
Subject: import(solutions): Add problem 6 .

---
 works/solutions/cpp/6.cpp | 56 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 works/solutions/cpp/6.cpp

(limited to 'works/solutions/cpp/6.cpp')

diff --git a/works/solutions/cpp/6.cpp b/works/solutions/cpp/6.cpp
new file mode 100644
index 0000000..f1d947c
--- /dev/null
+++ b/works/solutions/cpp/6.cpp
@@ -0,0 +1,56 @@
+#include <string>
+#include <iostream>
+
+using std::string;
+
+class Solution
+{
+public:
+    string convert(string s, int numRows)
+    {
+        if (numRows == 1)
+            return s;
+
+        const auto length = s.size();
+        const int count_per_group = numRows * 2 - 2;
+        string result;
+        result.reserve(length);
+        for (int row = 0; row < numRows; row++)
+        {
+            if (row == 0)
+            {
+                for (int p = 0; p < length; p += count_per_group)
+                    result += s[p];
+            }
+            else if (row == numRows - 1)
+            {
+                for (int p = row; p < length; p += count_per_group)
+                    result += s[p];
+            }
+            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)
+                    result += s[p];
+            }
+        }
+        return result;
+    }
+};
+
+int main()
+{
+    Solution s;
+
+    auto result1 = s.convert("PAYPALISHIRING", 3);
+    auto result2 = s.convert("PAYPALISHIRING", 4);
+    std::cout
+        << s.convert("A", 1) << '\n'
+        << result1 << '\n'
+        << "PAHNAPLSIIGYIR\n"
+        << result2 << '\n'
+        << "PINALSIGYAHRPI\n";
+    return 0;
+}
\ No newline at end of file
-- 
cgit v1.2.3