01_合并两个有序数组

• 合并两个有序数组

#include <iostream>
#include <vector>
using namespace std;

//给你两个按非递减顺序排列的整数数组nums1和nums2，另有两个整数m和n，
//分别表示nums1和nums2中的元素数目。
//
//请你合并nums2到nums1中，使合并后的数组同样按非递减顺序排列。
//注意：最终，合并后数组不应由函数返回，而是存储在数组nums1中。为了应对这种情况，
//nums1的初始长度为m + n，其中前m个元素表示应合并的元素，后n个元素为0 ，应忽略。nums2的长度为n。
class Solution
{
public:
	void merge(vector<int>& num1, int m, vector<int>& num2, int n);
};	
void Solution::merge(vector<int>& num1, int m, vector<int>& num2, int n) 
{
	int i = m - 1;
	int j = n - 1;
	int k = m + n - 1;
	while (i >= 0 && j >= 0)
	{
		if (num1[i] > num2[j])
		{
			num1[k--] = num1[i--];
		}
		else
		{
			num1[k--] = num2[j--];
		}
	}
	while (j >= 0)
	{
		num1[k--] = num2[j--];
	}
}
int main()
{
	vector<int> num1 = { 1, 2, 3, 0, 0, 0 };
	int m = 3;
	vector<int> num2 = { 2, 5, 6 };
	int n = 3;

	Solution sol;
	sol.merge(num1, m, num2, n);
	for (int num : num1)
	{
		cout << num << " ";
	}
	cout << endl;
	return 0;
}