LeetCode笔记：88. Merge Sorted Array

问题：

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.
Note:
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2. The number of elements initialized in nums1 and nums2 are m and n respectively.

大意：

给出两个排序好了的整型数组 nums1 和 nums2，将 nums2 合并到 nums1 中去成为一个排好序的数组。
注意：
你可以假设 nums1 有足够的空间（尺寸大于等于 m + n）来添加从 nums2 来的额外的元素。nums1 和 nums2 中的元素个数分别为 m 和 n。

思路：

这道题一开始并不想用最直接最笨的方法做，而是试图利用题目的条件来做。题目两个原本的数组都是排好序了的，题目没有返回值，因此只能在nums1中直接操作，不能放新的数组，题目额外给出了元素个数m和n，这都是暗示。不过最终也没做好，后来发现其实还是思路方向反了= =

这里先说最直接的办法，空间换时间，直接创建一个新数组，然后一个个从两个数组中按照顺序拿元素去比较大小放回nums1中，就得到一个排好序的数组了。

代码（Java）：

public class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        if (nums2.length == 0) return;
        
        int[] nums = Arrays.copyOfRange(nums1, 0, m);
        int i = 0;
        int j = 0;
        int k = 0;
        while (i < nums.length && j < nums2.length) {
            if (nums[i] <= nums2[j]) {
                nums1[k] = nums[i];
                i++;
            } else {
                nums1[k] = nums2[j];
                j++;
            }
            k++;
        }
        while (i < nums.length) {
            nums1[k] = nums[i];
            i++;
            k++;
        }
        while (j < nums2.length) {
            nums1[k] = nums2[j];
            j++;
            k++;
        }
        
        return;
    }
}

他山之石：

public class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int i=m-1, j=n-1, k=m+n-1;
        while (i>-1 && j>-1) nums1[k--]= (nums1[i]>nums2[j]) ? nums1[i--] : nums2[j--];
        while (j>-1)         nums1[k--]=nums2[j--];
        
        return;
    }
}

这个做法让我明白我的思路还是反了，之前一直想的还是顺序去找元素，还在想怎么处理原来位置的元素，这个方法直接从后面开始排，先放最后面的数，从大到小慢慢往前走，放的数一定不会覆盖原来nums1中的数，因为有足够的位置。如果nums2中的数先放完，那么剩下的nums1前面的数位置不用动，如果nums1的先放完，剩下还有一些nums2的数就直接放在前面去了。

此外，这里为了精简代码，使用了i--这种先取值然后减一的特性。

合集：https://github.com/Cloudox/LeetCode-Record

---

查看作者首页