LeetCode 88.合并两个有序数组

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]

思路

• 一
  • 先想到的是把nums1数组中长度为m的复制出来为nums1_copy
  • 讲nums1_copy，nums2采用双指针法依次从前往后遍历，把小的一个插入到nums1中
  • 时间复杂度O(n+m)
  • 空间复杂度 : O(m)

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int[] nums1_copy = new int[m];
        System.arraycopy(nums1, 0, nums1_copy, 0, m);
        int count = 0;
        int i = 0;
        int j = 0;
        while (i < m && j < n) {
            if (nums1_copy[i] > nums2[j]) {
                nums1[count] = nums2[j];
                j++;
            } else {
                nums1[count] = nums1_copy[i];
                i++;
            }
            count++;
        }
        while (i < m) {
            nums1[count] = nums1_copy[i];
            i++;
            count++;
        }
        while (j < n) {
            nums1[count] = nums2[j];
            j++;
            count++;
        }
    }
}

• 二
  • 因为内存消耗相对大，参考了其他的思路
  • 不用复制数组，采用从后往前的遍历，nums1从m处开始往前，和nums2从后往前的比较，大的数字从nums1末尾依次写入
  • 时间复杂度 : O(n + m)
  • 空间复杂度 : O(1)

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int i = m - 1;
        int j = n - 1;
        int k = m + n - 1;
        while (i >= 0 && j >= 0) {
            if (nums1[i] > nums2[j]) {
                nums1[k] = nums1[i];
                k--;
                i--;
            } else {
                nums1[k] = nums2[j];
                k--;
                j--;
            }
        }
        System.arraycopy(nums2, 0, nums1, 0, j + 1);
    }
}