python: add problem 88 and unittest

Author: supercoderhawk

Python/sln_1_100/solution_81_90.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+from typing import List
 from common_utils import ListNode
 
 
@@ -37,3 +38,33 @@ def partition(self, head: ListNode, x: int) -> ListNode:
             less_node.next = equal_more_than_head
             new_head = less_head
         return new_head
+
+    def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
+        """
+        88
+        Do not return anything, modify nums1 in-place instead.
+        """
+
+        merged_arr = []
+        if not m:
+            merged_arr = nums2
+        elif not n:
+            merged_arr = nums1
+        else:
+            i1 = i2 = 0
+            for _ in range(m + n):
+                if nums1[i1] < nums2[i2]:
+                    merged_arr.append(nums1[i1])
+                    i1 += 1
+                else:
+                    merged_arr.append(nums2[i2])
+                    i2 += 1
+                if i1 == m:
+                    merged_arr.extend(nums2[i2:n])
+                    break
+                if i2 == n:
+                    merged_arr.extend(nums1[i1:m])
+                    break
+
+        for i, val in enumerate(merged_arr):
+            nums1[i] = val

Python/sln_1_100/test_solution_81_90.py

@@ -3,13 +3,23 @@
 from sln_1_100.solution_81_90 import *
 from common_utils import *
 
+
 class Test_Solution_81_90(TestCase):
     def setUp(self) -> None:
         self.sln = Solution_81_90()
 
-
-
     def test_partition(self):
-        head = build_linked_list([1,4,3,2,5,2])
-        ret = self.sln.partition(head,3)
-        print_linked_list(ret)
+        head = build_linked_list([1, 4, 3, 2, 5, 2])
+        ret = self.sln.partition(head, 3)
+        print_linked_list(ret)
+
+    def test_merge(self):
+        arr1 = [1, 2, 3, 0, 0, 0]
+        self.sln.merge(arr1, 3, [2, 5, 6], 3)
+        self.assertEqual(arr1, [1, 2, 2, 3, 5, 6])
+        arr2 = [1]
+        self.sln.merge(arr2, 1, [], 0)
+        self.assertEqual(arr2, [1])
+        arr3 = [2, 0]
+        self.sln.merge(arr3, 1, [1], 1)
+        self.assertEqual(arr3, [1, 2])