【Leetcode】88.合并两个有序数组

88.合并两个有序数组

题目

给你两个有序整数数组 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]

方法(双指针)

使用两个指针,一个指向nums1中的元素,一个指向nums2中的元素

如果从前向后遍历的话,由于是从小到大重置数组nums1,向nums1中插入nums2元素的操作会造成后面元素的移动,增加时间复杂度。

因此我们需要从后向前遍历,也就是按元素从大到小重置数组nums1,每次比较指针p1和指针p2指向的nums1和nums2中的元素,每次将大的元素往nums1末尾放,直到放置好所有的元素。

1
2
3
4
5
6
7
8
9
10
11
12
public void merge(int[] nums1, int m, int[] nums2, int n) {
int p1 = m - 1;
int p2 = n - 1;
int index = m + n - 1;
while(p1 >= 0 && p2 >= 0){
nums1[index--] = nums1[p1] > nums2[p2] ? nums1[p1--] : nums2[p2--];
}
while(p1 >= 0)
nums1[index--] = nums1[p1--];
while(p2 >= 0)
nums1[index--] = nums2[p2--];
}
  • 时间复杂度:O(n)
  • 空间复杂度:O(1)