Leetcode刷题笔记 321. 拼接最大数

时间：2020年12月2日
知识点：贪心、合并、选取k个有序数使得字典序最大
题目链接：https://leetcode-cn.com/problems/create-maximum-number/

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

解法：

1. 思路还是比较清晰的
2. 长度为k，我们可以从第一个数组中取[max(0,k-num2len),min(k,num1len)]长度的顺序最长字符
3. 最少取 max(0,k-num2len) 最少取 k - numlen2 最多取k个 或者 num1len
4. 如何取这个长度下最大的顺序 字符串 这个是核心 也是就是 选取k个有序数使得字典序最大
5. 如果数组中还存在数 且剩下的数足够 且这个数比数组中的最后一位大 则一直弹出 最后放入
6. 两个字符串进行合并 merge
7. 最后比较 留下比较大的那个 compare

代码

#include <stdio.h>
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
class Solution {
public:
    vector<int> maxNumber(vector<int>& nums1, vector<int>& nums2, int k) {
        int m = nums1.size(),n = nums2.size();
        vector<int> ans(k,0);
        for(int i = max(0,k - n); i <= min(k,m); i++){
            vector<int> newnums1 = getmaxsubsequnce(nums1,i);
            vector<int> newnums2 = getmaxsubsequnce(nums2,k-i);
            vector<int> tmp = merge(newnums1,newnums2);
            if(compare(tmp,0, ans ,0) > 0)
                ans.swap(tmp);
        }
        return ans;
    }
    vector<int> getmaxsubsequnce(vector<int>& nums,int len){
        int n = nums.size();
        int pop = n - len;
        vector<int> ans;
        for(int i = 0;i < n;i++){
            while(!ans.empty() && nums[i] > ans.back() &&  pop-->0)
                ans.pop_back();
            ans.push_back(nums[i]);
        }
        ans.resize(len);
        return ans;
    }
    vector<int> merge(vector<int>& newnums1,vector<int>& newnums2){
        int x = newnums1.size(),y = newnums2.size();
        if(x==0)
            return newnums2;
        if(y==0)
            return newnums1;
        int newlen = x + y;
        vector<int> newnums(newlen,0);
        int idx1 = 0,idx2 = 0;
        for(int i = 0; i < newlen ;i++){
            if(compare(newnums1, idx1, newnums2, idx2) > 0){
                newnums[i] = newnums1[idx1++];
            }else{
                newnums[i] = newnums2[idx2++];
            }
        }
        return  newnums;
    }
    int compare(vector<int>& newnums1,int idx1,vector<int>& newnums2,int idx2){
        int x = newnums1.size(),y = newnums2.size();
        while(idx1 < x && idx2 < y){
            int diff = newnums1[idx1] - newnums2[idx2];
            if(diff != 0)
                return diff;
            idx1++;
            idx2++;
        }
        return (x - idx1) - (y- idx2);
    }
    
};
int main()
{
    vector<int> nums1{3,4,6,5};
    vector<int> nums2{9,2,3,5,8,3};
    int k = 5;
    Solution s;
    vector<int> ans = s.maxNumber(nums1, nums2, 5);
    for(int x :ans)
        cout<<x<<endl;
    return 0;
}

今天也是爱zz的一天哦！