LeetCode[321] Create Maximum Number

UsherChen 發(fā)布于2019-08-14 16:38 / 933人閱讀

摘要：算法復(fù)雜度思路貪心算法，先能組成的數(shù)的組合，然后針對每一個(gè)組合，考慮每一個(gè)數(shù)組能夠組成的最大的位或者位數(shù)。對不同組合生成的最大數(shù)進(jìn)行比較，得到所能得到的最大的值。代碼的方法去找這個(gè)數(shù)。

LeetCode[321] Create Maximum Number

Given two arrays of length m and n with digits 0-9 representing two

Create the maximum number of length k <= m + n from digits of

The relative order of the digits from the same array must be

Return an array of the k digits. You should try to optimize

time and space complexity.


Example 1: nums1 = [3, 4, 6, 5] nums2 = [9, 1, 2, 5, 8, 3] k = 5
return [9, 8, 6, 5, 3]
Example 2: nums1 = [6, 7] nums2 = [6, 0, 4] k = 5 return [6, 7, 6, 0,
4]
Example 3: nums1 = [3, 9] nums2 = [8, 9] k = 3 return [9, 8, 9]

Greedy算法

復(fù)雜度
O(k * (m + n)^3), O(k)

思路
貪心算法，先能組成k的數(shù)的組合，然后針對每一個(gè)組合(k = m + n)，考慮每一個(gè)數(shù)組能夠組成的最大的m位(或者n位)數(shù)。將這兩個(gè)數(shù)組形成的數(shù)進(jìn)行merge，得到針對這個(gè)組合下能得到的最大的數(shù)，然后記錄下來。對不同組合生成的最大數(shù)進(jìn)行比較，得到所能得到的最大的值。

代碼

    public int[] maxNumber(int[] nums1, int[] nums2, int k) {
        List list = new LinkedList<>();
        int[] ans = new int[k];
        for(int i = 0; i <= k; i ++) {
            if(i <= nums1.length && k - i <= nums2.length) {
                int[] res = merge(maxNum(nums1, i), maxNum(nums2, k - i), k);
                if(larger(res, ans, 0, 0)) ans = res;
            }
        }
        return ans;
    }
    
    // find the larger array;
    // return true when res value >= ans calue;
    public boolean larger(int[] res, int[] ans, int r, int a) {
        while(r < res.length && a < ans.length && res[r] == ans[a]) {
            r ++;
            a ++;
        }
        return a == ans.length || (r < res.length && res[r] > ans[a]);
    }
    
    // find the max number that can be formed in the nums, with i digits.
    // use greedy的方法去找這個(gè)數(shù)。
    public int[] maxNum(int[] nums, int k) {
        int[] res = new int[k];
        int j = 0, n = nums.length;
        for(int i = 0; i < nums.length; i ++) {
            // 因?yàn)橐ＷC順序，所以n - i > k - j, 同時(shí)，如果發(fā)現(xiàn)之前那一位比現(xiàn)在這一位要小的話，就replace成new value
            while(n - i + j > k && j > 0 && res[j - 1] < nums[i]) j --;
            if(j < k) res[j++] = nums[i];
        }
        return res;
    }
    
    // merge the two nums value that forms k digits;
    public int[] merge(int[] nums1, int[] nums2, int k) {
        int[] res = new int[k];
        int i = 0, j = 0;
        for(int index = 0; index < k; index ++) {
            // 關(guān)鍵比較的時(shí)候要這樣比較，不能直接取大的值，而是要取從這一位開始能夠組成最大的數(shù)的值。
            res[index] = larger(nums1, nums2, i, j) ? nums1[i ++] : nums2[j ++];
        }
        return res;
    }