(第9题是合并有序链表)
合并两个有序数组
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
说明:
初始化 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)
{
//用来找到在nums1中倒序添加元素的位置
int max_index = m + n - 1;
while (m && n) //当nums1和nums2数组中均存在元素还未排序时进行循环
{
//nums1的末位元素更大
if (nums1[m - 1] > nums2[n - 1])
{
nums1[max_index] = nums1[m - 1]; //将nums1末尾元素放在nums1的空白位置
max_index--; //两个指针前移
m--;
}
else
{
nums1[max_index] = nums2[n - 1];
max_index--;
n--;
}
}
while (n) //当nums1排完但nums2未排完时,说明nums1前面剩下的位置为空,直接将nums2剩下的元素放入即可
{
nums1[max_index] = nums2[n - 1];
max_index--;
n--;
}
}
};
思路:
- 从nums1和nums2末尾存在元素位置开始比较,用m和n分别代替指针,用来找到当前两数组进行比较的末尾元素,找到两数组中更大的末尾元素并放到nums1数组的末尾
- 当m或n为0时说明其中一个数组已经排序完成,退出上述循环。若m为0,表示nums1中原有的元素已经排完,此时若n不为0,则将剩余nums2中的元素依次放入nums1中即完成。若n为零,则表示nums2中所有元素已经插入nums1中,排序完成