- 题目描述:
给出两个长度分别是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]
3. 思路: 将 k 个数拆分成两部分: i 、k - i,统计所有组合方式中结果最大的vector,即为结果
备注:取k个最大数时,巧妙利用栈结构
4. 代码:
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) {
vector<int> res;
for(int i=0; i<=k; ++i){
if(i>nums1.size() || k-i>nums2.size()) continue;
vector<int> result = merge(findVector(nums1, i), findVector(nums2, k-i));
if(result>res) res = result;
}
return res;
}
vector<int> merge(vector<int> nums1, vector<int> nums2){
vector<int> res;
while(!nums1.empty() || !nums2.empty())
{
if(nums1>nums2){
res.push_back(nums1[0]);
nums1.erase(nums1.begin());
}
else{
res.push_back(nums2[0]);
nums2.erase(nums2.begin());
}
}
return res;
}
vector<int> findVector(vector<int> v, int k){
int m = v.size()-k;
vector<int> res;
for(auto c : v){
while(m && res.size() && res.back()<c){
res.pop_back();
--m;
}
res.push_back(c);
}
res.resize(k);
return res;
}
}