[LeetCode]321. Create Maximum Number

https://leetcode.com/problems/create-maximum-number/

两个数组nums1,nums2组成长度为k的数组,找出数组中元素相对位置不变的情况下,和数组所代表数的最大值



本题较难,分四步,看注释。两个数组合并成最大数组的方法是:找到第一个不相等的位置,数字大的所在的数组的第一个元素就是要取得数字,取完之后index后移。

public class Solution {
    public int[] maxNumber(int[] nums1, int[] nums2, int k) {
        int m = nums1.length;
        int n = nums2.length;
        int[] res = new int[k];
        // i的判断条件要注意,都是小于等于
        for (int i = Math.max(0, k - n); i <= k && i <= m; i++) {
            int[] candidates = merge(maxArray(nums1, i), maxArray(nums2, k - i));
            if (greater(candidates, 0, res, 0)) {
                res = candidates;
            }
        }
        return res;
    }
    // 找一个数组抽取k个数,相对位置不变,组成的最大数组
    private int[] maxArray(int[] nums, int k) {
        int[] res = new int[k];
        int n = nums.length;
        for (int i = 0, j = 0; i < n; i++) {
            while (n - i + j > k && j > 0 && res[j - 1] < nums[i]) {
                j--;
            }
            if (j < k) {
                res[j++] = nums[i];
            }
        }
        return res;
    }
    // 在两数组当前位置分别为i和j的情况下,后续位置nums1是否大于nums2
    private boolean greater(int[] nums1, int i, int[] nums2, int j) {
        while (i < nums1.length && j < nums2.length && nums1[i] == nums2[j]) {
            i++;
            j++;
        }
        return j == nums2.length || (i < nums1.length && nums1[i] > nums2[j]);
    }
    // 将两个数组合并成组成最大的和数组
    private int[] merge(int[] nums1, int[] nums2) {
        int k = nums1.length + nums2.length;
        int[] res = new int[k];
        for (int i = 0, j = 0, t = 0; t < k; t++) {
            res[t] = greater(nums1, i, nums2, j) ? nums1[i++] : nums2[j++];
        }
        return res;
    }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值