【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解

时间:2022-06-03 15:40:45

问题描述:

【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解问题解答: 

(1) 定义两个数组的元素个数 m ,n;

	vector<int> nums1 = { 1, 3, 5, 7 };
	int m = (int)nums1.size();

	vector<int> nums2 = { 2, 4, 6 };
	int n = (int)nums2.size();

(2)向数组 nums1 中push_back(0),使得数组nums1中的size为 m + n:

	for (int i = 0; i < nums2.size(); i++)
		nums1.push_back(0); 

【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解

(3)设定nums1的后四位数为1,3,5,7

		for (int i = n + m - 1; i >= n; i--)
			nums1[i] = nums1[i - n];

【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解

(4)初始化循环中的变量:

		int i = n;  // pointer for nums1 [n, n+m)
		int j = 0;  // pointer for nums2 [0, n)
		int k = 0;  // pointer merged nums1 [0, n+m)

如图:

【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解

(5) 创建循环体while (k < n + m)

①if (nums1[i] < nums2[j]) ,则nums1[k] = nums1[i];

如图:

【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解

然后 i++,k++:

【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解

 

 ② if (nums1[i] > nums2[j]) , 则 nums1[k] = nums2[j]

【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解

然后 k++,j++

【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解

 ③ if (j >= n) , 则nums1[k++] = nums1[i++]

【LeetCode】88.Merge Sorted Array(合并两个有序数组)-C++实现及详细图解

④ if (i >= n + m),则nums1[k++] = nums2[j++];

(6)完整的代码实现


#include <iostream>
#include <vector>
#include <cassert>

using namespace std;


void printVec(const vector<int>& vec) {
	for (int e : vec)
		cout << e << " ";
	cout << endl;
}


/// Standard merge process in merge sort
/// Time Complexity: O(n)
/// Space Complexity: O(n)

class Solution {
public:
	void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {

		assert(nums1.size() == m + n && nums2.size() == n);

		for (int i = n + m - 1; i >= n; i--)
			nums1[i] = nums1[i - n]; //输出:nums1[] = {1 3 5 1 3 5 7}

		int i = n;  // pointer for nums1 [n, n+m)
		int j = 0;  // pointer for nums2 [0, n)
		int k = 0;  // pointer merged nums1 [0, n+m)
		while (k < n + m) {
			if (i >= n + m)
				nums1[k++] = nums2[j++];
			else if (j >= n)
				nums1[k++] = nums1[i++];
			else if (nums1[i] < nums2[j])
				nums1[k++] = nums1[i++];
			else
				nums1[k++] = nums2[j++];
		}
	}
};


int main() {

	vector<int> nums1 = { 1, 3, 5, 7 };
	int m = (int)nums1.size();

	vector<int> nums2 = { 2, 4, 6 };
	int n = (int)nums2.size();

	for (int i = 0; i < nums2.size(); i++)
		nums1.push_back(0);       // 输出:nums1[] = {1 3 5 7 0 0 0}.

	Solution().merge(nums1, m, nums2, n);
	printVec(nums1);

	system("pause");
	return 0;
}

<全文完>

参考资料:

https://github.com/liuyubobobo/Play-Leetcode