合并两个有序数组
给你两个按 非递减顺序 排列的整数数组
nums1
和nums2
,另有两个整数m
和n
,分别表示nums1
和nums2
中的元素数目。请你 合并
nums2
到nums1
中,使合并后的数组同样按 非递减顺序 排列。注意:最终,合并后数组不应由函数返回,而是存储在数组
nums1
中。为了应对这种情况,nums1
的初始长度为m + n
,其中前m
个元素表示应合并的元素,后n
个元素为0
,应忽略。nums2
的长度为n
。示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3 输出:[1,2,2,3,5,6] 解释:需要合并 [1,2,3] 和 [2,5,6] 。 合并结果是 [1,2,2,3,5,6] ,其中斜体加粗标注的为 nums1 中的元素。示例 2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0 输出:[1] 解释:需要合并 [1] 和 [] 。 合并结果是 [1] 。示例 3:
输入:nums1 = [0], m = 0, nums2 = [1], n = 1 输出:[1] 解释:需要合并的数组是 [] 和 [1] 。 合并结果是 [1] 。 注意,因为 m = 0 ,所以 nums1 中没有元素。nums1 中仅存的 0 仅仅是为了确保合并结果可以顺利存放到 nums1 中。
方法一 双指针
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int i = 0, j = 0, cur = 0;
int[] arr = new int[m + n];
while (i < m || j < n) {
if (i == m) {
cur = nums2[j++];
} else if (j == n) {
cur = nums1[i++];
} else if (nums1[i] <= nums2[j]) {
cur = nums1[i++];
} else {
cur = nums2[j++];
}
arr[i + j - 1] = cur;
}
for (int k = 0; k != m + n; ++k) {
nums1[k] = arr[k];
}
}
}
方法二 逆序双指针
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int i = m - 1, j = n - 1, cur = 0;
int tail = m + n - 1;
while (i >= 0 || j >= 0) {
if (i == -1) {
cur = nums2[j--];
} else if (j == -1) {
cur = nums1[i--];
} else if (nums1[i] > nums2[j]) {
cur = nums1[i--];
} else {
cur = nums2[j--];
}
nums1[tail--] = cur;
}
}
}
逆序双指针解决了双指针需要额外开辟临时空间的问题,避免出现放入的元素将num1未比较的元素替换。
两种方法的时间复杂度:O(m + n)
方法一的空间复杂度: O(m+n) 方法二的空间复杂度: O(1)