给出两个长度分别是 m 和 n 的数组来表示两个大整数,数组的每个元素都是数字 0-9 。从这两个数组当中选出 k 个数字来创建一个最大数,其中 k 满足 k <= m + n。选出来的数字在创建的最大数里面的位置必须和在原数组内的相对位置一致。返回 k 个数的数组。你应该尽可能的去优化算法的时间复杂度和空间复杂度。写一个算法实现此功能。
格式:
输入第一行依次两个数组,最后输出选出的最大数的数组。
样例输入
nums1 = [ 3,4,6,5 ]
nums2 = [ 9,1,2,5,8,3 ]
k = 5
样例输出
[ 9,8,6,5,3 ]
解题代码如下:
package suanfa;
import java.util.Arrays;
public class MaxNum {
public static void main(String[] args) {
int[] array1 = new int[] {3, 4, 6, 5};
int[] array2 = new int[] {9, 1, 2, 5, 8, 3};
int k = 5;//数字长度
int start_index_array1 = -1;//记录array1最近(最大)被选中的元素的下标
int start_index_array2 = -1;//记录array2最近(最大)被选中的元素的下标
int[] maxNumArray = new int[k];//记录最大数的数字数组
int index = 0;//maxNumArray下标
while (index < k) {
int temp = start_index_array1;//暂存array1的上一次元素被选中的下标。在本次被选中的元素被丢弃之后要还原到上一次的下标位置
for (int i = start_index_array1 + 1; i < array1.length; i++) {
//如果maxNumArray当前index元素值小于array1[i],并且array1可取的剩余长度+array2可取的剩余长度 >= maxNumArray中还未赋值的剩余元素长度,
//则就采用此元素,start_index_array1更新为当前下标
if (maxNumArray[index] < array1[i] && ((array1.length - (i + 1)) + (array2.length - (start_index_array2 + 1))) >= (k - index - 1)) {
maxNumArray[index] = array1[i];
start_index_array1 = i;
}
}
for (int j = start_index_array2 +1; j < array2.length; j++) {
//如果maxNumArray当前index元素值小于array2[i],并且array1可取的剩余长度+array2可取的剩余长度 > maxNumArray中还未赋值的剩余元素长度,
//则就采用此元素,start_index_array2更新为当前下标。start_index_array1还原为上一次的下标
if (maxNumArray[index] < array2[j] && ((array1.length - (temp + 1)) + (array2.length - (j + 1))) >= (k - index - 1)) {
maxNumArray[index] = array2[j];
start_index_array1 = temp;
start_index_array2 = j;
}
}
index++;
}
System.out.println(Arrays.toString(maxNumArray));
}
}
输出结果如下:
[9, 8, 6, 5, 3]