python: add problem 6 and unittest

Author: supercoderhawk

Python/sln_1_100/solution_1_10.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+class Solution_1_10(object):
+    def convert(self, s: str, numRows: int) -> str:
+        """
+        6
+        :param s:
+        :param numRows:
+        :return:
+        """
+        if numRows == 1:
+            return s
+
+        item = []
+        arr = []
+        i = 0
+        while s:
+            item.append(s[0])
+            i += 1
+            s = s[1:]
+            if i == 2 * numRows - 2:
+                arr.append(item)
+                item = []
+                i = 0
+        if item:
+            arr.append(item)
+        final_res = []
+        for row in range(numRows):
+            for idx in range(len(arr)):
+                if len(arr[idx]) < row + 1:
+                    continue
+                final_res.append(arr[idx][row])
+                if 0 < row < numRows - 1:
+                    if len(arr[idx]) < 2 * numRows - 1 - row:
+                        continue
+                    final_res.append(arr[idx][2 * numRows - 2 - row])
+        return ''.join(final_res)

Python/sln_1_100/test_solution_1_10.py

@@ -0,0 +1,12 @@
+# -*- coding: utf-8 -*-
+from unittest import TestCase
+from sln_1_100.solution_1_10 import Solution_1_10
+
+
+class Test_Solution_1_10(TestCase):
+    def setUp(self) -> None:
+        self.sln = Solution_1_10()
+
+    def test_convert(self):
+        self.assertEqual(self.sln.convert("PAYPALISHIRING", 3), "PAHNAPLSIIGYIR")
+        self.assertEqual(self.sln.convert("AB", 1), "AB")