Leetcode 88. 合并两个有序数组

Tags: Two Pointer,双指针

题目

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]

提示:

What if you fill the longer array from the end instead of start ?

题目分析

根据提示用num1做输出数组，且num1的空间等于两个数组元素总和，我们可以从num1的尾部开始合并，用双指针法。

1. 定义num1的最后一个元素索引j，定义num2的最后一个元素索引k，反向变量变量num1；
2. 当j,k大于0,且 nums2[k] > nums1[j], num1[i]=nums2[k]；
3. 如不满足上述条件，且j≥0,则赋值num1[i]=nums1[j]；
4. num1被pop结束后，如果num2还有元素，则按顺序把num2合并到num1。
5. 重复相同过程直到循环结束。

复杂度分析

• 时间复杂度：O(n)， 假设数组总共有m+n 个元素，至少遍历m+n 步。
• 空间复杂度：O(1)。

Java

public static void merge(int[] nums1, int m, int[] nums2, int n) {

        int j = m - 1;
        int k = n - 1;
       
        
        for (int i = nums1.length - 1; i >= 0; i --) {
            if (k >= 0 && j >= 0 && nums2[k] > nums1[j] ) {
                nums1[i] = nums2[k];
                k --;
            } else if (j >= 0) {
                nums1[i] = nums1[j];
                j --;
            } else if (k >= 0) {
                nums1[i] = nums2[k];
                k --;
            }
        }
    }