LeetCode88 Merge Sorted Array 合并有序序列 C++

最新推荐文章于 2022-03-20 23:42:47 发布

tigerhuli

最新推荐文章于 2022-03-20 23:42:47 发布

阅读量134

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/m0_37518259/article/details/81023805

版权

LeetCode 专栏收录该内容

503 篇文章 3 订阅

订阅专栏

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:

The number of elements initialized in nums1 and nums2 are m and n respectively.
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.

Example:

Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6],       n = 3

Output: [1,2,2,3,5,6]

题源：here；完整实现：here

思路：

同样的思路两种方案：1 从前往后；2 从后往前。都是one-pass的方案，但是第二种不需额外申请空间。

1 从前往后

void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
	vector<int> nums;
	int i = 0, j = 0;
	while(i < m && j < n) {
		if (nums1[i] <= nums2[j]){
			nums.push_back(nums1[i]); i++;
		}
		else{
			nums.push_back(nums2[j]); j++;
		}
	}
	if (j < n) nums.insert(nums.end(), nums2.begin() + j, nums2.begin()+n);
	else if(i < m) nums.insert(nums.end(), nums1.begin() + i, nums1.begin()+m);
	nums1 = nums;
}

2 从后往前

void merge2(vector<int>& nums1, int m, vector<int>& nums2, int n){
	m--; n--;
	for (int i = m + n + 1; i >= 0; i--){
		if (m >= 0 && n >= 0 && nums1[m] >= nums2[n]) nums1[i] = nums1[m--];
		else if (m >= 0 && n >= 0 && nums1[m] < nums2[n]) nums1[i] = nums2[n--];
		else if (m >= 0) nums1[i] = nums1[m--];
		else if (n >= 0) nums1[i] = nums2[n--];
	}
}