leetcode 321 Create Max Number

时间:2023-03-10 05:13:13
leetcode 321 Create Max Number

leetcode 321 Create Max Number

greedy的方法,由于有两个数组,我们很自然的想到从数组1中选i个数,数组2中选k-i个数,这样我们只需要遍历max(0, k-数组2长度n) ~ 数组1长度,然后保存合并i和k-i这两个部分之后得到的最大值即可。 那么还剩下这几个问题:

1) 从一个数组中不改变元素顺序地选出i个元素,使他们顺序排列代表的十进制数最大。例:[1,4,3,6,2], i = 3, 结果为[4,6,2]

2) 如何合并数组1的i个元素和数组2的k-i个元素

3) 比较两个数组所代表的十进制数的大小

第一个问题,我们需要使用栈的概念,遍历数组,每次比较栈顶元素与当前元素i,若i比栈顶元素大,则pop栈顶,继续比较,直到栈空或者i比栈顶元素小。注意,我们还需要时刻检测,i之后的元素是否足够补上pop出栈的空缺,若不够则不能再pop。例:[1,4,3,6,2], 当index为3时,栈中有[4,3],6和3比,大于,弹出3,继续比4,大于,这时候却不能弹出4,因为6后面只有一个数,不够补上pop2次的空缺。

第二个问题,分别用两个index i, j指向数组1和2,比较他们指向的元素,将大的元素放入新的大小为k的数组。注意,如何比较呢?可能指向元素相等,可能某一个数组已经到头了。详细过程在第三个问题列出

第三个问题,如何比较?本题有两个地方需要使用比较,第一个就是主函数中比较不同i值导致的不同长度为k的数组,这个较为简单,因为两个数组长度一致,只需要用两个index,i,j分别遍历两个数组即可,返回nums1[i]和nums2[j]中较大的那一个,若相等则i++,j++,若i=j=k-1还是相等,说明两个数组代表的十进制数相同,返回哪一个都可以。第二个地方就是上一个问题中提到的,比较两个index指向元素的大小,一部分与之前解法相同,元素相等就继续比较后面的元素,不过有一点需要区别的,这里的两个数组可能是不同长度的,所以有可能i或j中某一个指向的位置没有数组元素,即这个数组已经到头了,这种情况,显然就可以返回另一个数组的index指向的元素。

代码如下:

 class Solution {
public int[] maxNumber(int[] nums1, int[] nums2, int k) {
int len1 = nums1.length, len2 = nums2.length;
int[] res = new int[k]; for(int i=0; i<=k; i++){
if((k - i) <= len2 && i <= len1){
int[] tmp = new int[k];
tmp = merge(singleNum(nums1, i), singleNum(nums2, k-i));
if(compare(tmp, 0, res, 0))
res = tmp;
}
} return res;
} public int[] singleNum(int[] nums, int k){
int len = nums.length;
int[] res = new int[k];
int j = 0; for(int i=0; i<len; i++){ while(len-i+j > k && j > 0 && res[j-1] < nums[i])
j--;
if(j < k)
res[j++] = nums[i];
}
return res; } public int[] merge(int[] nums1, int[] nums2){
int k = nums1.length + nums2.length;
int[] res = new int[k]; for(int i=0, j=0, idx = 0; idx < k; idx++){
res[idx] = compare(nums1, i, nums2, j) ? nums1[i++] : nums2[j++];
}
return res; } private boolean compare(int[] nums1, int idx1, int[] nums2, int idx2){
while(idx1 < nums1.length && idx2 < nums2.length && nums1[idx1] == nums2[idx2]){
idx1++;
idx2++;
}
return idx2 == nums2.length || (idx1 < nums1.length && nums1[idx1] > nums2[idx2]); }
}