LeetCode 88: 合并两个有序数组

[guodongxiaren]

https://leetcode-cn.com/problems/merge-sorted-array/

给你两个有序整数数组 nums1 和 nums2，请你将 nums2 合并到 nums1 中，使 nums1 成为一个有序数组。

说明:

初始化 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]

[guodongxiaren]

这是一个简单题【O(n)辅助空间复杂度】

注意细节。要不停求min，不是求max。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {

        int head[2] = {0};
        int N = m+n;
        vector<int> res(N);
        for (int i = 0; i < N; ++i) {
            int min = INT_MAX;
            int index = -1;
            if (head[0] < m && nums1[head[0]] < min) {
                min = nums1[head[0]];
                index = 0;
            }

            if (head[1] < n && nums2[head[1]] < min) {

                min = nums2[head[1]];
                index = 1;
            }
            res[i] = min;

            head[index]++;
        }
        for (int i = 0; i < N; i++) {
            nums1[i] = res[i];
        }
        //swap(nums1, res);
    }
};

最后如果用swap反而更慢：

[guodongxiaren]

O(1)辅助空间复杂度

如果原地排序不用开辟新数组呢？

可以用双指针！逆序

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {   
        int i = m-1;
        int j = n-1;
        for (int k = m + n -1; k >=0; --k) {
            if (i >= 0 && j >= 0) {
                if (nums1[i] > nums2[j]) {
                    nums1[k] = nums1[i--];
                } else {
                    nums1[k] = nums2[j--];
                }
            } else if (i >= 0) {
                nums1[k] = nums1[i--];
            } else if (j >= 0) {
                nums1[k] = nums2[j--];
            }
        }
    }
};

所谓双指针不是说两个数组两个指针，而是说一个数组用两个指针。和compact算法类似，或者说和memmove的内部类似。通过逆序来规避元素覆盖。