leetcode321. 拼接最大数

题目：321. 拼接最大数

class Solution {
    public int[] maxNumber(int[] nums1, int[] nums2, int k) {
        // 取值范围： max(0, k-nums2.length) <= t <= min(k, nums1.length), t表示从nums1取几个数
        int m = Math.min(k, nums1.length); 
        int n = Math.max(0, k-nums2.length);
        int[] res = new int[k];
        for(int i=n; i<=m; i++)
        {
            // 分治
            int[] maxArr1 = getMaxArr(nums1, i); // 从nums1中抽取i个值，保持相对顺序不变
            int[] maxArr2 = getMaxArr(nums2, k-i); // 从nums2中抽取k-i个值，保持相对顺序不变
            // 合并子问题
            int[] arr = merge(maxArr1, maxArr2, k); // 将两者结合，保持顺序不变，使得数组最大。
            // 与当前最大数做比较
            for(int j=0; j<k; j++)
            {
                if(arr[j] == res[j]) continue;
                if(arr[j] > res[j]) res = arr;
                break;
            }
        }
        return res;
    }

    public int[] getMaxArr(int[] nums, int k)
    {
        if(k == 0) return new int[0];
        int cursor = -1;
        int[] arr = new int[k];
        for(int i=0; i<nums.length; i++)
        {
            while(cursor >= 0 && arr[cursor] < nums[i] && k-cursor <= nums.length-i) // 
                cursor--;
            if(cursor < k-1)
                arr[++cursor] = nums[i];
        }
        return arr;
    }
    
    public int[] merge(int[] nums1, int[] nums2, int k)
    {
        int i1 = 0, i2 = 0, i3 = 0;
        int[] arr = new int[k];
        while(i1 < nums1.length && i2 < nums2.length)
            arr[i3++] = compare(nums1, i1, nums2, i2)? nums1[i1++] : nums2[i2++];
        while(i1 < nums1.length)
            arr[i3++] = nums1[i1++];
        while(i2 < nums2.length)
            arr[i3++] = nums2[i2++];
    
        return arr;
    }

    // 将两个数组结合起来的时候如果遇到相同的值，那么下移判断，直到遇到第一个不相同的值，然后进行比较判断。
    public boolean compare(int[] nums1, int i1, int[] nums2, int i2)
    {
        while(i1 < nums1.length && i2 < nums2.length)
        {
            if(nums1[i1] == nums2[i2])
            {
                i1++;
                i2++;
                continue;
            }
            return nums1[i1] > nums2[i2];
        }
        return i1 < nums1.length;

    }
}