题目链接:88. 合并两个有序数组
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
初始化 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]
程序说明:
第一种做法:双指针算法,从前往后。但是要用到额外的空间(临时数组),时间复杂度:O(m + n)。
第二种做法:双指针从后往前,不需要额外空间,时间复杂度O(m + n)。
代码如下:
class Solution {
public void merge(int[] A, int m, int[] B, int n) {
int tmp[] = new int[A.length];
int i = 0, j = 0, k = 0;
while(i < m && j < n) {
if(A[i] <= B[j]) tmp[k++] = A[i++];
else tmp[k++] = B[j++];
}
while(i < m) tmp[k++] = A[i++];
while(j < n) tmp[k++] = B[j++];
for(i = 0, j = 0; i < m + n; i++, j++)
A[i] = tmp[j];
}
}
class Solution {
public static void merge(int[] nums1, int m, int[] nums2, int n) {
int i = m - 1, j = n - 1, k = n + m - 1;
while(i >= 0 && j >= 0) {
if(nums1[i] <= nums2[j]) nums1[k--] = nums2[j--];
else nums1[k--] = nums1[i--];
}
while(j >= 0) nums1[k--] = nums2[j--];
}
}