欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

88. 合并两个有序数组

程序员文章站 2024-02-25 09:20:04
...

给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。

说明:

初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:

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

输出: [1,2,2,3,5,6]

思路:
经典归并算法,懒得动脑,没省空间,下一个思路是个抖机灵方法。

class Solution {
public:
	void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) 
	{
		vector<int>temp;
		vector<int>n1(nums1.begin(), nums1.begin() + m);
		vector<int>n2(nums2.begin(), nums2.begin() + n);
		int p1 = 0, p2 = 0;
		while (p1<n1.size()&&p2<n2.size())
		{
			if (n1[p1] > n2[p2])
				temp.push_back(n2[p2++]);
			else
				temp.push_back(n1[p1++]);
		}
		while (p1 < n1.size())
			temp.push_back(n1[p1++]);
		while (p2 < n2.size())
			temp.push_back(n2[p2++]);
		nums1 = temp;
	}
};

思路2:
鉴于本题的描述,我们可以把nums2塞到nums1后面,然后再把nums1排序即可,(正经脸)

class Solution {
public:
	void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
	{
		for (int i = 0; i < n; ++i)
		{
			nums1.at(m + i) = nums2.at(i);
		}
		sort(nums1.begin(), nums1.begin() + m + n);
	}
};