LeetCode面試題 10.01. 合併排序的數組

題目描述

給定兩個排序後的數組 A 和 B，其中 A 的末端有足夠的緩衝空間容納 B。 編寫一個方法，將 B 合併入 A 並排序。
初始化 A 和 B 的元素數量分別爲 m 和 n。

示例

輸入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6], n = 3

輸出:
[1,2,2,3,5,6]

說明:
A.length == n + m

思路1

將B賦值到A的緩衝區，再進行快排：

代碼1

    def merge(self, A, m, B, n):
        """
        :type A: List[int]
        :type m: int
        :type B: List[int]
        :type n: int
        :rtype: None Do not return anything, modify A in-place instead.
        """

        A[m:] = B
        A.sort()
        return A

時間複雜度分析：即快排的時間複雜度:O((m+n)log(m+n))
空間複雜度分析：快排：O(log(m+n))

思路2：

雙指針，遍歷兩個列表，並比較首元素大小，將較小的元素放入第三個列表當中。

代碼2：

    def merge2(self,A,m,B,n):
        '''

        :param A: 1,2,3,0,0,0
        :param m:
        :param B: 2,5,6
        :param n:
        :return:
        '''
        C = []
        a, b = 0, 0
        while a < m or b < n:
            if a == m:
                C.append(B[b])
                b += 1
            elif b == n:
                C.append(A[a])
                a += 1
            elif A[a] < B[b]:
                C.append(A[a])
                a += 1
            else:
                C.append(B[b])
                b += 1
  		A[:] = C
        return A

時間複雜度分析：遍歷兩列表長度即O(m+n)
空間複雜度分析：第三個列表C長度爲m+n,佔用O(m+n)

氣溫回升了，越來越暖和，新冠病毒退散~
想我的梨子餃子~