【LeetCode】321. Create Maximum Number 拼接最大数（Hard）（JAVA）

【LeetCode】321. Create Maximum Number 拼接最大数（Hard）（JAVA）

题目地址: https://leetcode.com/problems/create-maximum-number/

题目描述：

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]

题目大意

给定长度分别为 m 和 n 的两个数组，其元素由 0-9 构成，表示两个自然数各位上的数字。现在从这两个数组中选出 k (k <= m + n) 个数字拼接成一个新的数，要求从同一个数组中取出的数字保持其在原数组中的相对顺序。

求满足该条件的最大数。结果返回一个表示该最大数的长度为 k 的数组。

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

解题方法

1. 求出数组 nums 的 k 个最大序列
2. 合并 nums1 的 i 个最大序列和 nums2 的 j 个最大序列（其中 i + j = k）成 k 长度的数组
3. 对 num1 和 num2 遍历，分别找出 i 和 j 的长度的最大序列（其中 i + j = k） ，然后合并比较最大值
4. note: 合并的时候可以自己定义排序方式

class Solution {
    public int[] maxNumber(int[] nums1, int[] nums2, int k) {
        int[] res = new int[k];
        for (int i = Math.max(0, k - nums2.length); i <= nums1.length && i <= k; i++) {
            int[] temp = merge(getNum(nums1, i), getNum(nums2, k - i));
            if (compare(res, temp, 0, 0)) {
                res = temp;
            }
        }
        return res;
    }

    public int[] getNum(int[] nums, int k) {
        if (k == 0) return new int[]{};
        int[] res = new int[k];
        int pos = 0;
        for (int i = 0; i < nums.length; i++) {
            while (pos > 0 && res[pos - 1] < nums[i] && (nums.length - i) > (k - pos)) {
                pos--;
            }
            if (pos < k) {
                res[pos] = nums[i];
                pos++;
            }
        }
        return res;
    }

    public int[] merge(int[] nums1, int[] nums2) {
        int[] res = new int[nums1.length + nums2.length];
        int i = 0;
        int j = 0;
        while (i < nums1.length || j < nums2.length) {
            if (i >= nums1.length) {
                res[i + j] = nums2[j];
                j++;
            } else if (j >= nums2.length) {
                res[i + j] = nums1[i];
                i++;
            } else if (compare(nums1, nums2, i, j)) {
                res[i + j] = nums2[j];
                j++;
            } else {
                res[i + j] = nums1[i];
                i++;
            }
        }
        return res;
    }

    public boolean compare(int[] nums1, int[] nums2, int i, int j) {
        for (; i < nums1.length && j < nums2.length; i++, j++) {
            if (nums1[i] == nums2[j]) continue;
            return nums1[i] < nums2[j];
        }
        return j < nums2.length;
    }
}

执行耗时:8 ms,击败了97.00% 的Java用户
内存消耗:39.4 MB,击败了25.67% 的Java用户

欢迎关注我的公众号，LeetCode 每日一题更新