My code:
public class Solution {
public int[] maxNumber(int[] nums1, int[] nums2, int k) {
if (nums1 == null || nums2 == null || nums1.length + nums2.length < k) {
return null;
}
int[] ret = null;
int max = 0;
for (int i = 0; i <= k; i++) {
int[] n1 = pick(nums1, i);
int[] n2 = pick(nums2, k - i);
if (n1 == null || n2 == null) {
continue;
}
int[] m = merge(n1, n2);
ret = getSum(m, ret);
}
return ret;
}
private int[] pick(int[] nums, int k) {
if (nums.length < k) {
return null;
}
int[] ret = new int[k];
int cursor = 0;
for (int i = 0; i < nums.length; i++) {
while (cursor > 0 && nums[i] > ret[cursor - 1] && cursor + nums.length - i - 1 >= k) {
cursor--;
}
if (cursor < k) {
ret[cursor] = nums[i];
cursor++;
}
}
return ret;
}
private int[] merge(int[] n1, int[] n2) {
int l1 = n1.length;
int l2 = n2.length;
int[] ret = new int[l1 + l2];
int i = 0;
int j = 0;
for (int k = 0; k < ret.length; k++) {
if (bigger(n1, i, n2, j)) {
ret[k] = n1[i];
i++;
}
else {
ret[k] = n2[j];
j++;
}
}
return ret;
}
private boolean bigger(int[] n1, int i, int[] n2, int j) {
while (i < n1.length && j < n2.length) {
if (n1[i] < n2[j]) {
return false;
}
else if (n1[i] > n2[j]) {
return true;
}
else {
i++;
j++;
}
}
return i != n1.length;
}
private int[] getSum(int[] nums, int[] ret) {
if (ret == null) {
return nums;
}
if (nums.length > ret.length) {
return nums;
}
else if (nums.length < ret.length) {
return ret;
}
else {
for (int i = 0; i < nums.length; i++) {
if (nums[i] > ret[i]) {
return nums;
}
else if (nums[i] < ret[i]) {
return ret;
}
}
return ret;
}
}
public static void main(String[] args) {
Solution test = new Solution();
int[] nums1 = new int[]{2, 5, 6, 4, 4, 0};
int[] nums2 = new int[]{7, 3, 8, 0, 6, 5, 7, 6, 2};
int[] ret = test.maxNumber(nums1, nums2, 15);
for (int i = 0; i < ret.length; i++) {
System.out.print(ret[i] + "_");
}
}
}
这道题目思考了一会儿，觉得没有思路。然后看了答案的提示。
然后尝试着自己写出来。写了很久，最后卡在了
merge函数的地方。
慢慢说来。
reference:
https://www.hrwhisper.me/leetcode-create-maximum-number/
我自己也很自然的想到，可以写一个循环，然后分割。
arr1 取出 i
arr2 取出 k - i
如何让值最大
那么这个大问题就可以分解为多个子问题，然后求出最大解就可以了。
下面分析这个问题：
arr1 取出 i
arr2 取出 k - i
如何让值最大
他又可以分成三个问题：
- 从 arr 取出 i 位，如果保证取出的是最大的
- 现在有 arr1, arr2, 如何merge 成最大的 (这个问题最难，然后一开始我以为最简单，以为就是 Merge sort)
- 如何计算合并后的arr的值 (这里会存在over flow,因为，arr可能有40位或者更多，Long都不会够，所以得用其他的比较方法)
首先解决问题1：
其实就是一个栈的问题。
当 stack.size() < k 的时候，我不断地插入新的元素。
当 = k 时，我可以判断当前元素nums[i] 是否大于栈顶元素：
stack.peek()
错！ 一开始我也这么想的，这是不对的！
我维持一个栈，当栈空时，直接插，不空时，
判断 当前元素nums[i] 是否大于栈顶元素：
stack.peek()
如果小于，
如果栈size小于k，那就直接插，i继续走。
否则就直接忽略。
如果大于，那么就要复杂的思考下了。
按道理说，就该把栈顶弹出，然后把这个插进去。
但如果剩下的元素个数如果已经小于我们还需要的个数时，就凑不满了，那么这个时候就不能弹出，只能忽略，继续往前走。
简而言之，
如果 指针 i 指向 nums[] 的当前元素
指针cursor指向 ret[] 的当前元素 （就是栈）
if (cursor > 0 && nums[i] > ret[cursor - 1] && cursor + nums.length - i >= k)
{
cursor--;
}
这里，cursor - 1 指的是栈顶，
cursor = stack size
所以如果弹出，然后插入新的元素，
我们还需要的元素个数是：
k - cursor
我们还剩下可供选择的元素个数是：
nums.length - i
k - cursor <= nums.length - i
=>
cursor + nums.length - i >= k
那么这个时候，我们就可以弹出了。
以此作为一个while循环，不断地更新这个栈，直到是当前最优值。
那么第一个问题解决了。
第二个问题，如何合并。
我一开始以为是简单地merge sort，后来发现不是。
[3, 3, 4]
[3, 3]
当 arr1[i] == arr2[j]时，在merge sort中是可以任意先让哪个指针先走的。
这里的话，其实不是merge sort，只是将两个数组合并，保持他们原有的顺序，并且使得值最大。
那么应该是，
3, 3, 4, 3, 3
所以，需要判断哪个数组更大，更大的话就走哪个。
也就是说， arr1[i] > or < arr2[j] 时，一切照常
如果 arr1[i] == arr2[j] 时，需要一直遍历这两个数组，直到找到一个元素，两者不等，或者其中一个数组被穷尽。
第二个问题也算是解决了。
第三个问题，如果比较大小。
最后还是打算用字符串来做这件事，方便安全。
也解决了。
这道题目的考虑点还是很多的。我觉得有用的是第一二两个问题的答案。
如何从 arr[] 中选出 k个，顺序组成一个数字，最大
如何 merge arr1[] and arr2[] ，保持他们的顺序，使其组成的数字最大。
Anyway, Good luck, Richardo! -- 08/25/2016