【LeetCode】321. 拼接最大数 结题报告 (C++)

原题地址：https://leetcode-cn.com/problems/create-maximum-number/submissions/

题目描述：

给定长度分别为 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]
 

解题方案：

本题使用的方法是贪心算法和动态规划。

在主函数中，将k个元素分配到给定的两个数组，通过遍历所有的情况，找到最大排序的数组，在这里使用了贪心算法。

maxVector（）函数用来获取数组的最大子序列，mergeVector（）函数根据原有的两个数组从大到小进行合并，并保持原有数组的顺序。

mergeVector（）函数的代码实现技巧值得学习，每次取数组地址。

动态规划的思想我好像没有看出来。

代码：

class Solution {
public:
    vector<int> maxNumber(vector<int>& nums1, vector<int>& nums2, int k){
        int m = nums1.size(), n = nums2.size();
        vector<int> res;
        for(int i = max(0, k - n); i <= min(k, m); i ++)
            res = max(res, mergeVector(maxVector(nums1, i), maxVector(nums2, k - i)));
        return res;
    }
    vector<int> maxVector(vector<int> nums, int k){
        int drop = nums.size() - k;
        vector<int> res;
        for(int num : nums){
            while(drop && res.size() && res.back() < num){
                res.pop_back();
                -- drop;
            }
            res.push_back(num);
        }
        res.resize(k);
        return res;
    }
    vector<int> mergeVector(vector<int> nums1, vector<int> nums2){
        vector<int> res;
        while(nums1.size() + nums2.size()){
            vector<int> &tmp = nums1 > nums2 ? nums1 : nums2;
            res.push_back(tmp[0]);
            tmp.erase(tmp.begin());
        }
        return res;
    }
};