合并两个有序数组（C++解法）

题目

给你两个按非递减顺序排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。

请你合并 nums2 到 nums1 中，使合并后的数组同样按非递减顺序排列。

注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。

C++代码

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

/*
* 定义数组合并排序函数
* 定义两个指针数组，倒序遍历并插入比较后的元素
*/
void sort_array(vector<int>& nums1, int m, vector<int>& nums2, int n) {
	int len1 = m - 1, len2 = n - 1, len = m + n - 1;

	while (len1 >= 0 && len2 >= 0) {

		if (nums1[len1] > nums2[len2]) {
			nums1[len--] = nums1[len1--];
		}
		else {
			nums1[len--] = nums2[len2--];
		}
	}
	while (len2 >= 0) {
		nums1[len--] = nums2[len2--];
	}
}

//定义数组输出函数
void print_array(vector<int>& nums1) {

	for (int i = 0; i < nums1.size(); ++i){
		cout << nums1[i] << " ";
	}
}

int main() {
	int m, n, number;
	vector<int> nums1, nums2;

	cout << "nums1 = ";
	while (cin >> number) {
		nums1.push_back(number);
		if (cin.get() == '\n'){
			break;
		}
	}

	cout << "nums2 = ";
	while (cin >> number) {
		nums2.push_back(number);
		if (cin.get() == '\n') {
			break;
		}
	}

	n = nums2.size();
	m = nums1.size() - n;

	sort_array(nums1, m, nums2, n);
	cout << "Merged nums1: ";
	print_array(nums1);

	return 0;
}

分析

• 算法部分：定义一个 sort_array() 函数，合并排序两个指针数组 nums1, nums2，并将有序数组放入 nums1 中，其中我们采用倒序遍历的方式进行，减少运行时间。将 nums1 的最后一个非零元素和 nums2 最后一个元素作比较，得出更大的元素放入 nums1 的最后一个索引位置中，循环至条件终止。
• 输出部分：定义一个 print_array() 数组输出函数，将得出的 nums1 顺序输出。
• 主函数部分：我们写上测试案例，调用 sort_array() 函数和 print_array() 函数。

问题

因为语法还不够熟悉，我遇到的问题主要在输入输出方面：vector 容器定义的 nums1、nums2 怎样输入任意长度数组，循环输出每个元素并且范围在 nums1.size() 内，防止数组越界，也没有注意到题目中给的 nums1.size() = m + n, nums2.size() = n，一直出现运行报错。