leetcode#321 Create Maximum Numbers

Given two arrays of length m and n with digits 0-9 representing two numbers. Create the >maximum number of length k <= m + n from digits of the two. The relative order of the digits >from the same array must be preserved. Return an array of the k digits. You should try to >optimize your 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]

这道题是一个典型的dp问题。可以考虑从nums1中取i个数构成最大值,从nums2中取k-i个数取最大值。容易求得,i的下限是max(k-len(nums2),0),上限是min(len(nums1),0)

首先是在每个数列中构造最大数列。需要比较遍历数列并比较当前值与所构造数列的尾部值的大小。需要注意的是,如果数列尾部是递减的,可能导致所构造数列元素个数超过k个,所以需要将多余的元素减去。

其次是将两个最大数列合并。这里要特别注意到考虑两个元素相等的情况,要继续比较后续元素的大小。事实上,在C++的STL中直接比较vector的大小就可以实现。

class Solution {
public:
    vector<int> maxVector(vector<int>& list, int k) {
        vector<int> result;
        if(k==0) return result;
        int drop = list.size() - k;
        for(int i=0; i<list.size(); i++) {
            while(result.size() && drop>0 && list[i]>result.back()) {
                result.pop_back();
                drop--;
            }
            result.push_back(list[i]);
        }
        result.resize(k);
        return result;
    }

    vector<int> merge(vector<int>& list1, vector<int>& list2) {
        vector<int> result;
        while(list1.size()+list2.size()>0) {
            vector<int>& tmp = list1>list2?list1:list2;
            result.push_back(tmp[0]);
            tmp.erase(tmp.begin());
        }
        return result;
    }

    vector<int> maxNumber(vector<int>& nums1, vector<int>& nums2, int k) {
        vector<int> result;
        int L = max(k-int(nums2.size()), 0);
        int R = min(int(nums1.size()), k);
        for (int i = L; i <= R; i++) {
            vector<int> v1 = maxVector(nums1, i);
            vector<int> v2 = maxVector(nums2, k-i);
            result = max(result, merge(v1, v2));
        }
        return result;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值