leetCode321：拼接最大数

目录

一、题目描述

二、解题思路

三、代码实现

---

一、题目描述

给定长度分别为 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]

示例 2:

输入:
nums1 = [6, 7]
nums2 = [6, 0, 4]
k = 5
输出:
[6, 7, 6, 0, 4]

示例 3:

输入:
nums1 = [3, 9]
nums2 = [8, 9]
k = 3
输出:
[9, 8, 9]

二、解题思路

首先关于题目思路 ：

1. 首先需要分别循环从两个数组里取数，假设nums1里面取了x个，nums2里取了y个，必须满足x+y=k，取数需要满足所得结果最大，并且相对顺序不变，如果经验丰富的话，就能想到单调栈的结构，单调栈内数据单调的性质可以满足题目中最大或者最小的要求，并且栈的出入规则也能做到相对顺序不变。
2. 取到数后还需要将分别取到的数进行合并，使得合并的结果最大。这里合并非常类似归并排序的合并过程，哪边的数大就先合并哪边的数。
3. 取整个循环过程中最大的结果作为最终结果返回。

其次，这道题还有几个需要注意的细节：

1. 在用单调栈取数的时候需要维持栈中的数据量等于你需要的长度；
2. 需要自定义比较函数，比较两个数组哪个更大；
3. 在进行合并的时候，如果出现了相等的情况，需要选择两个子数组更大的哪一边先合并，这样才能保证结果最大。

三、代码实现

#include <bits/stdc++.h>
using namespace std;

vector<int> getNumsFromVector(vector<int>& nums, int k) {
	int n = nums.size();
	if (n == 0) return {};
	vector<int> stack;
	stack.push_back(nums[0]);
	for (int i = 1; i < n; ++i) {
		//需要保证在遍历结束后栈中的数据至少有k个，
		//所以当当前的数比栈顶小要出栈时，要考虑后续的数字数量够不够用
		while (!stack.empty() && nums[i] > stack.back() && (n - i > k - (int)stack.size())) {
			stack.pop_back();
		}
		stack.push_back(nums[i]);
	}
	//如果长度超了k，需要截断
	if (stack.size() > k) {
		vector<int> res(stack.begin(), stack.begin() + k);
		return res;
	}
	return stack;
}
//如果nums1大返回1，如果nums2大或相等返回2
int maxNumsVector(vector<int>& nums1, int index1, vector<int>& nums2, int index2) {
	int m = nums1.size(), n = nums2.size();
	if (m == 0) return 2;
	if (n == 0) return 1;
	int p1 = index1, p2 = index2;
	for (; p1 < m && p2 < n; ++p1, ++p2) {
		if (nums1[p1] < nums2[p2]) {
			return 2;
		}
		if (nums1[p1] > nums2[p2]) {
			return 1;
		}
	}
	if (p1 < m) {
		return 1;
	} else {
		return 2;
	}
}
vector<int> merge(vector<int>& nums1, vector<int>& nums2) {
	int m = nums1.size(), n = nums2.size();
	if (m == 0) return nums2;
	if (n == 0) return nums1;
	vector<int> res(m + n, 0);
	int p1 = 0, p2 = 0;
	int i = 0;
	while (p1 < m && p2 < n) {
		if (nums1[p1] < nums2[p2]) {
			res[i++] = nums2[p2++];
			//相等的时候怎么接需要考虑
			//应该接后面的数组更大的那一边
		} else if (nums1[p1] == nums2[p2]) {
			if (maxNumsVector(nums1, p1, nums2, p2) == 1) {
				res[i++] = nums1[p1++];
			} else {
				res[i++] = nums2[p2++];
			}
		}
		else{
			res[i++] = nums1[p1++];
		}
	}
	while (p1 < m) {
		res[i++] = nums1[p1++];
	}
	while (p2 < n) {
		res[i++] = nums2[p2++];
	}
	return res;
}
vector<int> maxNumber(vector<int>& nums1, vector<int>& nums2, int k) {
	int m = nums1.size();
	int n = nums2.size();
	//数组1能取到的长度的最小值begin和最大值end
	int begin = max(0, k - n), end = min(k, m);
	vector<int> ansNums(k, 0);
	for (int i = begin; i <= end ; ++i) {
		vector<int> tempNums1 = getNumsFromVector(nums1, i);
		vector<int> tempNums2 = getNumsFromVector(nums2, k - i);
		vector<int> mergeRes = merge(tempNums1, tempNums2);
		if (maxNumsVector(ansNums, 0, mergeRes, 0)== 2) {
			ansNums = mergeRes;
		}
	}
	return ansNums;
}
int main() {
	vector<int> nums1 = { 2, 5, 6, 4, 4, 0 };
	vector<int> nums2 = { 7, 3, 8, 0, 6, 5, 7, 6, 2 };
	nums2 = maxNumber(nums1, nums2, 15);
	for (int i = 0; i < nums2.size(); i++) {
		cout << nums2[i] <<" ";
	}
	return 0;
}