问题

Description:Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:

  • The number of elements initialized in nums1 and nums2 are m and n respectively.
  • You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.

Example:

Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
Output: [1,2,2,3,5,6]
  • Level: Easy

解题思路

  初看题目很简单，而且一看就是归并排序。所以开始想着就是顺着归并排序的思路来做，也就是从头开始
遍历；会发现需要移动数组nums1，导致时间复杂度超过 O(n+m)，不满足归并排序。后面想想，居然形成了思维定
势。为了满足O(n+m),可以从后面开始遍历，只是我们学习的归并排序都是从头开始，而不是从尾部开始。一旦由从
尾部开始遍历的x想法，问题就解决了。具体见下面的详细代码。

详细代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
public void merge(int[] nums1, int m, int[] nums2, int n) {
    int i = m-1;
    int j = n-1;
    int total = m + n - 1;
    while (i >= 0 && j >= 0) {
        if (nums1[i] >= nums2[j]) {
            nums1[total--] = nums1[i--];
        } else {
            nums1[total--] = nums2[j--];
        }
    }
    while (j >= 0) {
        nums1[total--] = nums2[j--];
    }
}

分析

  • 时间复杂度为O(n+m),其中n为排序数组的长度；