给定长度分别为 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]
来源:力扣(LeetCode)
题目分析
①.该题的本质即为从两个数组中挑选 k 个元素,并且使拼接后的结果的字典序最大。
②.要使数组的字典序最大,则需要保证从左到右的每一个位置的值尽可能的大,即单调递减。
③.枚举所有 k 的组合情况,分别从两个数组中取 i、j 个元素构成的最大数组,然后再进行合并,最终保留最大值。
④.在挑选指定个数的元素的过程中,若当前供挑选的元素数量大于需要的个数,则进行取舍操作,以使每一个位置的值尽可能的大。
⑤.合并过程中,分别从头到尾遍历两个数组并比较当前元素优先取较大值作为合并结果;若当前位置值相等,则需要进行进行下一位的比较,以确保下一位可选的元素值最大。
字典序比较算法
①.序列的字典序比较:从第一位开始分别比较对应的元素值,若分出大小则返回结果,否则继续进行下一位的大小。
②.vector 的默认比较操作即为字典序比较。
③.STL 中的 lexicographical_compare() 算法可以按字典序比较由开始和结束迭代器定义的两个序列。
单调栈
①.单调栈:就是栈内元素单调按照递增(递减)顺序排列的栈。
②.单调栈的实现:可以使用 vector 来实现栈操作,vector 可以在末尾快速的添加和删除元素,并且方便栈内元素的读取。
vector<int> result;//单调递减栈
for( auto &num : nums)//遍历每个元素,进行添加操作
{
// 当前比较元素大于栈顶元素,则出栈、直到满足单调性
while( result.size() > 0 num > result.back() )
{
result.pop_back();//末尾删除元素
}
// 当前元素入栈
result.push_back(num) ;
}
代码示例
class Solution {
public:
vector<int> maxNumber(vector<int>& nums1, vector<int>& nums2, int k) {
vector<int> result;
int size1 = nums1.size();
int size2 = nums2.size();
for( int i = 0;i <= size1;i++ )//从第一个数组中取 i 个元素
{
if( i > k ) break;//最多取 k 个
if( size2 >= k -i )//第二个数组中能取够剩余元素
{
vector<int> vec1 = getMax(nums1,i);
vector<int> vec2 = getMax(nums2,k-i);
vector<int> res = meger2vec(vec1,vec2);
result = max(result,res);
}
}
return result;
}
//从数组中取 k 位数字,并保持相对顺序,使其构成的数字最大
vector<int> getMax(vector<int>& nums1,int k)
{
//要使数字最大,就使从左到右每一位尽可能的最大,即单调递减
vector<int> result;//返回结果
int size = nums1.size();
int remain = size -k;//允许丢弃的元素个数
for( auto &num : nums1)
{
//如果下个元素值比当前位置的值大,则可以选择舍弃当前元素
//但可舍弃的元素个数小于 1 时,表示不能再舍弃元素了,只能添加
while( result.size() > 0 && remain > 0 && num > result.back() )
{
remain--;//允许舍弃的元素个数减一
result.pop_back();//末尾删除元素
}
if( result.size() < k ) //未取够k个,添加到末尾
{
result.push_back(num) ;
}
else//已经取够了k个,直接丢弃
{
remain--;
}
}
return result;
}
//合并两个递减数组,使结果保存递减,并且构成的数字最大
vector<int> meger2vec(vector<int>& nums1,vector<int>& nums2)
{
//当两个位置数值一样时,要接着比较下一位,以使下一位尽可能的大
vector<int> result;
int size1 = nums1.size();
int size2 = nums2.size();
int i = 0;
int j = 0;
while( i < size1 && j < size2)
{
//哪个大取哪个
if( vector<int>(nums1.begin()+i,nums1.end()) > vector<int>(nums2.begin()+j,nums2.end()) )
{
result.push_back(nums1[i]);
i++;
}
else
{
result.push_back(nums2[j]);
j++;
}
}
//拷贝剩余元素到结果数组
copy(nums1.begin()+i,nums1.end(),back_inserter(result));
copy(nums2.begin()+j,nums2.end(),back_inserter(result));
return result;
}
};