给定长度分别为 m 和 n 的两个数组,其元素由 0-9 构成,表示两个自然数各位上的数字。现在从这两个数组中选出 k (k <= m + n) 个数字拼接成一个新的数,要求从同一个数组中取出的数字保持其在原数组中的相对顺序。
求满足该条件的最大数。结果返回一个表示该最大数的长度为 k 的数组。
说明: 请尽可能地优化你算法的时间和空间复杂度。
示例 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]
通过次数7,131
提交次数20,649
思路:贪心,即将大问题转化为小问题,从数组num1中取i个值,保持i个值的相对位置不变,从数组num2中取k-i个值,也保持这k-i个值的相对位置不变。然后将二者结合,也保持顺序不变,使得数组最大。有一点值得注意,将两个数组结合起来的时候,如果遇到相同的值,则下移判断,直到遇到第一队不相同的值,然后进行比较。
class Solution {
public static int[] getMax(int[] nums,int num){
int[] max = new int[num];
int index=0;
for(int j=0;j<nums.length;j++){
while(index>0&&(nums.length-j)>(num-index)&&nums[j]>max[index-1]){
index--;
}
if(index<num){
max[index++]=nums[j];
}
}
return max;
}
public static int[] merge(int[] nums1,int[] nums2){
int[] sum = new int[nums1.length+nums2.length];
for(int i=0,a=0,b=0;i<sum.length;i++){
sum[i]=greater(nums1,nums2,a,b)?nums1[a++]:nums2[b++];
}
return sum;
}
public static boolean greater(int[] nums1,int[] nums2,int i,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]);
}
public int[] maxNumber(int[] nums1, int[] nums2, int k) {
int[] max=null;
for(int i=0;i<=k&&i<=nums1.length;i++){
int j=k-i;
if(j>nums2.length)continue;
int[] sum1=getMax(nums1,i);
int[] sum2=getMax(nums2,j);
int[] sum3=merge(sum1,sum2);
if(null ==max||greater(sum3,max,0,0)){
max=sum3;
}
}
return max;
}
}