321. 拼接最大数

题目描述：给定长度分别为 m 和 n 的两个数组，其元素由 0-9 构成，表示两个自然数各位上的数字。现在从这两个数组中选出 k (k <= m + n) 个数字拼接成一个新的数，要求从同一个数组中取出的数字保持其在原数组中的相对顺序。
求满足该条件的最大数。结果返回一个表示该最大数的长度为 k 的数组。
说明: 请尽可能地优化你算法的时间和空间复杂度。
解题思路：（1）分别求出两个数组的最大子序列，使得两个子序列长度之和为k,两个数组的子序列长度组合最大有(k+1)中情况，最大子序列可以用单调栈解决，再保证选取长度的情况下，使得数字单调递减排列（2）将两个最大子序列进行合并，从左到右依次比较两个数组当前的数字，如果某一个数组的数更大，就将这个数添加大最终结果中，再依次比较后面的数，如果相等，则需要比较后面的数字，选择后面的数更大的数组的当前数添加到结果中，原则是如果后面的数更大，那么当前这个数选取之后，在下一次就有可能取到更大的数（3）从所有的组合中选择最大的结果返回，代码如下：

class Solution:
    def maxNumber(self, nums1: List[int], nums2: List[int], k: int) -> List[int]:
        def maxsubsequence(s, k):
            res = [0] * k
            top = -1
            remain = len(s) - k
            for i, num in enumerate(s):
                while top >= 0 and res[top] < num and remain > 0:
                    top -= 1
                    remain -= 1
                if top < k -1:
                    top += 1
                    res[top] = num
                else:
                    remain -= 1
            return res
        def compare(s1, idx1, s2, idx2):
            len1, len2 = len(s1), len(s2)
            while(idx1 < len1 and idx2 < len2):
                diff = s1[idx1] - s2[idx2]
                if diff != 0:
                    return diff
                idx1 += 1
                idx2 += 1
            return (len1 - idx1) - (len2 - idx2)
        
        def merge(s1, s2):
            len1, len2 = len(s1), len(s2)
            idx1 = idx2 = 0
            res = []
            while(idx1 < len1 and idx2 < len2):
                if compare(s1, idx1, s2, idx2) > 0:
                    res.append(s1[idx1])
                    idx1 += 1
                else:
                    res.append(s2[idx2])
                    idx2 += 1
            for i in range(idx1, len1):
                res.append(s1[i])
            for i in range(idx2, len2):
                res.append(s2[i])
            return res
        m, n = len(nums1), len(nums2)
        start, end = max(0, k - n), min(k, m)
        maxsequence = [0] * k
        for i in range(start, end+1):
            s1 = maxsubsequence(nums1, i)
            s2 = maxsubsequence(nums2, k-i)
            curmax = merge(s1, s2)
            if compare(curmax, 0, maxsequence, 0) > 0:
                maxsequence = curmax
        return maxsequence