假设我们有两个长度为m和n的数组,数字0-9代表两个数字。我们必须从两个数字中创建小于m + n的最大长度k。我们必须记住,必须保留同一数组中数字的相对顺序。我们必须找到k个数字的数组。因此,如果输入像[3,4,7,5]和[9,1,3,5,8,4],并且k = 5,则答案将是[9,8,7,5,4 ]。
为了解决这个问题,我们将遵循以下步骤-
定义一个函数mergeThem()
,它将使用一个数组nums1,一个数组nums2,
定义数组ret
i:= 0,j:= 0,n:= nums1的大小,m:= nums2的大小
而(i <n或j <m),做-
在ret的末尾插入nums2 [j]
(将j增加1)
在ret的末尾插入nums1 [i]
(将i增加1)
如果调用功能Greater(nums1,nums2,i,j)为true,则-
除此以外
返回ret
定义一个函数modify()
,它将采用数组v,k,
定义一个堆栈st
定义数组ret
对于初始化i:= 0,当i <v的大小时,更新(将i增加1),执行-
将x插入st
从st删除元素
x:= v [i]
while(st不为空,且st的顶部元素<x,st的大小+ v的大小– i – 1> = k),执行-
如果st <k的大小,则-
而(st不为空),-
在ret的末尾插入st的top元素
从st删除元素
反转数组ret
返回ret
定义一个函数greater()
,它将采用数组a,数组b,i,j,
而(i <a的大小,j <b的大小,并且a [i]与b [j]相同),则执行-
将i加1并将j加1
当j == b的大小或i <a的大小且a [i]> b [j]时返回true
从主要方法执行以下操作:
定义数组ret
n:= nums1的大小
m:= nums2的大小
对于初始化i:= 0,当i <= k时,更新(将i增加1),执行-
定义一个候选数组= mergeThem(modify(nums1,i),Modify(nums2,k-i))
如果更大(候选,ret,0、0)为真,则-
ret:=候选人
如果i <= n并且(k-i)<= m,则-
返回ret
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h> using namespace std; void print_vector(vector<auto> v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << v[i] << ", "; } cout << "]"<<endl; } class Solution { public: vector<int> mergeThem(vector<int> nums1, vector<int> nums2) { vector<int> ret; int i = 0; int j = 0; int n = nums1.size(); int m = nums2.size(); while (i < n || j < m) { if (greater(nums1, nums2, i, j)) { ret.push_back(nums1[i]); i++; } else { ret.push_back(nums2[j]); j++; } } return ret; } vector<int> modify(vector<int>& v, int k) { stack<int> st; vector<int> ret; for (int i = 0; i < v.size(); i++) { int x = v[i]; while (!st.empty() && st.top() < x && st.size() + (v.size() - i) - 1 >= k) { st.pop(); } if (st.size() < k) st.push(x); } while (!st.empty()) { ret.push_back(st.top()); st.pop(); } reverse(ret.begin(), ret.end()); return ret; } bool greater(vector<int>& a, vector<int>& b, int i, int j) { while (i < a.size() && j < b.size() && a[i] == b[j]) i++, j++; return j == b.size() || (i < a.size() && a[i] > b[j]); } vector<int> maxNumber(vector<int>& nums1, vector<int>& nums2, int k) { vector<int> ret; int n = nums1.size(); int m = nums2.size(); for (int i = 0; i <= k; i++) { if (i <= n && (k - i) <= m) { vector<int> candidate = mergeThem(modify(nums1, i), modify(nums2, k - i)); if (greater(candidate, ret, 0, 0)) { ret = candidate; } } } return ret; } }; main() { Solution ob; vector<int> v = { 3, 4, 7, 5 }, v1 = { 9, 1, 3, 5, 8, 4 }; print_vector(ob.maxNumber(v, v1, 5)); }
{ 3, 4, 7, 5 } { 9, 1, 3, 5, 8, 4 } 5
输出结果
[9, 8, 7, 5, 4, ]