题目描述:
给定长度分别为 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]
反思错误:
看的别人的算法,思路还是挺好理解的,也不存在什么错误不错误。还是注意小细节吧,,比如问什么要<=,为什么&&
解题思路:
把题分解来看:
①求[两个数组的 k 位长最大数]
②求[两数组抽出 k 位数 合并排列的 k 位最大数]
③求[一数组取 i 位数] 和 [二数组取 k-i 位数]
④求[一数组 i 位长最大数] 和 [二数组 k-i 位长最大数]
这样我们就能求出最大数。
ps:我的代码 变量命名都很长(不是很规范)还是有注释的。。
Java代码:
//321. 拼接最大数
//执行用时 : 27 ms, 在Create Maximum Number的Java提交中击败了66.18% 的用户
//内存消耗 : 40.3 MB, 在Create Maximum Number的Java提交中击败了0.00% 的用户
class Solution {
//寻找当前数组k位长的最大数,k<=nums.length
private int[] maxInNums(int[] nums,int k){
int[] max = new int[k];//作为返回的k位长 数
int len = nums.length;//nums的长度
//i用来指向nums[],,j用来指向max[]
for(int i=0,j=0;i<len;++i){
//当 j>0 且 max[]剩余容量小于nums[]的剩余容量 且 max存的最后一个值小于当前位置的nums值
// ,,就回退j的位置,把最大的nums值放进max
while(j>0 && k-j<len-i && max[j-1]<nums[i]) --j;
if(j<k)max[j++]=nums[i];
}
return max;
}
//返回nusm1Max大于nums2Max吗?
private boolean greater(int[] nums1Max,int i,int[] nums2Max,int j){
int lenNums1Max = nums1Max.length;
int lenNums2Max = nums2Max.length;
//当nums1Max[i] 与 nums2Max[j] 的值相同的时候,往后推
while(i<lenNums1Max && j<lenNums2Max && nums1Max[i]==nums2Max[j]){++i;++j;}
//true的情况:nums2Max到最后都与nums1Max相同 或者 nums1Max大于2Max
return j==lenNums2Max || (i<lenNums1Max && nums1Max[i]>nums2Max[j]);
}
//合并 nums1的max数组 与 nums2的max数组 --------》成一个大数组
private int[] merge(int[] nums1Max, int[] nums2Max){
int lenCurRes = nums1Max.length+nums2Max.length;
int[] curRes = new int[lenCurRes];
for(int i=0,j=0,m=0;m<lenCurRes;++m){
curRes[m] = greater(nums1Max,i,nums2Max,j)?nums1Max[i++]:nums2Max[j++];
}
return curRes;
}
//main
public int[] maxNumber(int[] nums1, int[] nums2, int k) {
int[] res = null;
for(int i=Math.max(k-nums2.length,0);i<=Math.min(nums1.length,k);++i){
int[] merge = merge(maxInNums(nums1,i),maxInNums(nums2,k-i));
res = (res==null || greater(merge,0,res,0))?merge:res;
}
return res;
}
}