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;
}
};