leetcode 321. 拼接最大數

給定長度分別爲 m 和 n 的兩個數組，其元素由 0-9 構成，表示兩個天然數各位上的數字。如今從這兩個數組中選出 k (k <= m + n) 個數字拼接成一個新的數，要求從同一個數組中取出的數字保持其在原數組中的相對順序。

求知足該條件的最大數。結果返回一個表示該最大數的長度爲 k 的數組。

說明: 請儘量地優化你算法的時間和空間複雜度。

示例 1:

輸入:
nums1 = [3, 4, 6, 5]
nums2 = [9, 1, 2, 5, 8, 3]
k = 5
輸出:
[9, 8, 6, 5, 3]
示例 2:

輸入:
nums1 = [6, 7]
nums2 = [6, 0, 4]
k = 5
輸出:
[6, 7, 6, 0, 4]
示例 3:

輸入:
nums1 = [3, 9]
nums2 = [8, 9]
k = 3
輸出:
[9, 8, 9]

學習一下別人的想法(開始想用rmp作,在相等的時候會超時, 233)

 1 def maxNumber(self, nums1, nums2, k):
 2 
 3     def prep(nums, k):
 4         drop = len(nums) - k
 5         out = []
 6         for num in nums:
 7             while drop and out and out[-1] < num:
 8                 out.pop()
 9                 drop -= 1
10             out.append(num)
11         return out[:k]
12 
13     def merge(a, b):
14         return [max(a, b).pop(0) for _ in a+b]
15 
16     return max(merge(prep(nums1, i), prep(nums2, k-i))
17                for i in range(k+1)
18                if i <= len(nums1) and k-i <= len(nums2))

// 這代碼看着真舒服