leetcode笔记—生成最大的 数组Create Maximum Number

给定两给数组，从中选取K个使这K个在原始数组中的顺序不变，使数组最大

思路：涉及到三个函数：1.取数组中k个最大的元素，顺序不变；

                                             2.合并两个数组，顺序不变，使最大；

                                             3.比较两个数组的大小；

class Solution {
public:
    
vector<int> maxArray(vector<int> nums, int k)
{  //取数组中K个最大的元素
     vector<int> res(k);
     stack<int> stack;  //栈，先进后出
     for(int i = 0 ;i < nums.size(); i++)
     {
         while(!stack.empty() && nums.size() - i + stack.size() > k && nums[i] > stack.top())
             stack.pop(); 
         if(stack.size() < k)
             stack.push(nums[i]);
     }
    while(!stack.empty())
    {
        res[k-1] = stack.top();
        stack.pop();
        k--;
    }
    return res;
}

vector<int> merge(vector<int> nums1, vector<int> nums2, int k)
{    //合并两个数组
    vector<int> res(k);
    for(int i = 0,j= 0,l =0; l < k; l++)
    {
        res[l] = greaterArray(nums1,nums2,i,j) ? nums1[i++]:nums2[j++];
    }
    return res;
}

bool greaterArray(vector<int> nums1, vector<int> nums2, int i, int j)
{
    while(i < nums1.size() && j < nums2.size() && nums1[i] == nums2[j])
    {
        i++;
        j++;
    }

   return j == nums2.size() || (i < nums1.size() && nums1[i] > nums2[j]);
}

vector<int> maxNumber(vector<int>& nums1, vector<int>& nums2, int k) 
{
    vector<int> ans(k,INT_MIN);
    for(int i = max(0,k-(int)nums2.size()); i <= k && i <= nums1.size(); i++)
    {
        vector<int> tmp = merge(maxArray(nums1,i),maxArray(nums2,k-i),k);
        if(greaterArray(tmp,ans,0,0))
        ans = tmp;
    }
    return ans;
        
}
};