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]);
            
    }
}