321. 拼接最大数

321. 拼接最大数

给定长度分别为 m 和 n 的两个数组，其元素由 0-9 构成，表示两个自然数各位上的数字。现在从这两个数组中选出 k (k <= m + n) 个数字拼接成一个新的数，要求从同一个数组中取出的数字保持其在原数组中的相对顺序。求满足该条件的最大数。结果返回一个表示该最大数的长度为 k 的数组。

 

示例 1:

输入:
nums1 = [3, 4, 6, 5]
nums2 = [9, 1, 2, 5, 8, 3]
k = 5
输出:
[9, 8, 6, 5, 3]

示例 2:

输入:
nums1 = [6, 7]
nums2 = [6, 0, 4]
k = 5
输出:
[6, 7, 6, 0, 4]

示例 3:

输入:
nums1 = [3, 9]
nums2 = [8, 9]
k = 3
输出:
[9, 8, 9]

---

基本思路：分治思想和单调栈，该问题需要分为两步，首先是分别从两个数组分别抽取k1,k2个数字，要保证抽取的数字最大；紧接着，根据大小合并序列

• 怎么从数组中抽取k个数字，保证抽取的k个数字组成的序列最大？采用降序栈，当出现升序时，出栈
• 怎么合并？合并时要注意当两个当前字符相等时，要比较后面的字符串。

    vector<int> getKmaxSequent(vector<int> &nums, int k) {
        if (nums.size() <= k)
            return nums;

        int drop = nums.size() - k, top = -1;
        vector<int> ans(nums);

        for (int i = 0; i < nums.size(); i++) {
            while (top != -1 && drop > 0 && nums[i] > ans[top]) {   //降序栈，当升序时，出栈
                drop--;
                top--;
            }
            ans[++top] = nums[i];
        }

        ans.resize(k);
        return ans;

    }

    vector<int> merge(vector<int> &t1, vector<int> &t2) {
        vector<int>  ans;
        int m = t1.size(), n = t2.size();
        if(m==0)
            return t2;
        if(n==0)
            return t1;
        int i = 0, j = 0;
        while(i+j<m+n){
            compare(t1,i,t2,j)?ans.push_back(t1[i++]):ans.push_back(t2[j++]); //当前两个字符相等时，要根据后面的字符比较大小。
        }
        return ans;

    }

    bool compare(vector<int> &newNums,int pos1, vector<int> &oldNums,int pos2) {
        int len1=newNums.size(),len2=oldNums.size();
        int i=pos1,j=pos2;
        while(i<len1&&j<len2){
            if(newNums[i]!=oldNums[j])
                return newNums[i]>oldNums[j];
            else{
                i++;
                j++;
            }
        }
        return len1-i>len2-j;
    }
    vector<int> maxNumber(vector<int>& nums1, vector<int>& nums2, int k) {
        vector<int> res;
        int m = nums1.size(), n = nums2.size();
        int start = max(0, k - n), end = min(k, m);
        for (int i = start; i <= end; i++) {
            vector<int> t1 = getKmaxSequent(nums1, i);
            vector<int> t2 = getKmaxSequent(nums2, k - i);
            vector<int> t = merge(t1, t2);//合并时，注意比较的是字符串，而不仅仅是字符
            if (compare(t,0, res,0) > 0)
                res.swap(t);
        }
        return res;
    }