[Swift]LeetCode321. 拼接最大数 | Create Maximum Number

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（shanqingyongzhi）
➤博客园地址：山青咏芝（https://www.cnblogs.com/strengthen/）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：https://www.cnblogs.com/strengthen/p/10260640.html 
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

Given two arrays of length m and n with digits 0-9 representing two numbers. Create the maximum number of length k <= m + n from digits of the two. The relative order of the digits from the same array must be preserved. Return an array of the k digits.

Note: You should try to optimize your time and space complexity.

Example 1:

Input:
nums1 = [3, 4, 6, 5]
nums2 = [9, 1, 2, 5, 8, 3]
k = 5
Output:
[9, 8, 6, 5, 3]

Example 2:

Input:
nums1 = [6, 7]
nums2 = [6, 0, 4]
k = 5
Output:
[6, 7, 6, 0, 4]

Example 3:

Input:
nums1 = [3, 9]
nums2 = [8, 9]
k = 3
Output:
[9, 8, 9]

---

说明: 请尽可能地优化你算法的时间和空间复杂度。

示例 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]

---

380 ms

class Solution {
    func maxNumber(_ nums1: [Int], _ nums2: [Int], _ k: Int) -> [Int] {
        let m = nums1.count
        let n = nums2.count
        
        var res = [Int]()
        
        let c = max(0, k-n)
        
        for i in c...min(k, m) {
            let r1 = maxNumArr(nums1, i)
            let r2 = maxNumArr(nums2, k-i)
            let tmp = maxNums(r1, r2, k)
            if isGreater(tmp, res, 0, 0) {
                res = tmp
            }
        }
        
        return res
    }
    
    func maxNumArr(_ nums : [Int], _ k : Int) -> [Int] {
        var res = [Int]()
        for i in 0..<nums.count {
            let num = nums[i]
            while !res.isEmpty && num > res.last! && nums.count + res.count > k + i {
                res.removeLast()
            }
            res.append(num)
            continue
        }
        return res
    }
    
    func maxNums(_ num1 : [Int], _ num2 : [Int],_ k : Int) -> [Int] {
        var res = [Int]()
        var i = 0 , j = 0
        for _ in 0..<k {
            if isGreater(num1, num2, i, j) {
                res.append(num1[i])
                i+=1
            }else {
                res.append(num2[j])
                j+=1
            }
        }
        
        return res
    }
    
    func isGreater(_ nums1: [Int], _ nums2 : [Int], _ i : Int, _ j : Int) -> Bool {
        var i = i , j = j
        while i < nums1.count , j < nums2.count && nums1[i] == nums2[j] {
            i+=1
            j+=1
        }
        
        return j == nums2.count || (i < nums1.count && nums1[i] > nums2[j])
    }
}