给出两个长度分别是m和n的数组来表示两个大整数,数组的每个元素都是数字0-9。从这两个数组当中选出k个数字来创建一个最大数,其中k满足k <= m + n。选出来的数字在创建的最大数里面的位置必须和在原数组内的相对位置一致。返回k个数的数组。你应该尽可能的去优化算法的时间复杂度和空间复杂度。
样例:
给出 nums1 = [3, 4, 6, 5], nums2 = [9, 1, 2, 5, 8, 3], k = 5
返回 [9, 8, 6, 5, 3]
给出 nums1 = [6, 7], nums2 = [6, 0, 4], k = 5
返回 [6, 7, 6, 0, 4]
给出 nums1 = [3, 9], nums2 = [8, 9], k = 3
返回 [9, 8, 9]
思路:
由于取k个数,可以分别在nums1,nums2中取k1,k2个数,k1+k2=k;
计算在nums中取k个数,组合成最大数,返回vector;
将得到的vector合并,遍历不同的组合情况,取最大值。
#ifndef C552_H
#define C552_H
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
/*
* @param nums1: an integer array of length m with digits 0-9
* @param nums2: an integer array of length n with digits 0-9
* @param k: an integer and k <= m + n
* @return: an integer array
*/
vector<int> maxNumber(vector<int> &nums1, vector<int> &nums2, int k) {
// write your code here
vector<int> res;
for (int i = 0; i <= k; ++i)
{
if (i > nums1.size() || k - i > nums2.size())
continue;
if (mergeVector(findVector(nums1, i), findVector(nums2, k - i)) > res)
res = mergeVector(findVector(nums1, i), findVector(nums2, k - i));
}
return res;
}
vector<int> findVector(vector<int> v, int k)
{
vector<int> res;
int num = v.size() - k;
for (auto c : v)
{
while (num&&!res.empty() && res.back() < c)
{
res.pop_back();
--num;
}
res.push_back(c);
}
res.resize(k);
return res;
}
vector<int> mergeVector(vector<int> v1, vector<int> v2)
{
vector<int> res;
while (!v1.empty() || !v2.empty())
{
if (v1 > v2)
{
res.push_back(v1[0]);
v1.erase(v1.begin());
}
else
{
res.push_back(v2[0]);
v2.erase(v2.begin());
}
}
return res;
}
};
#endif