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;
}
}