合并两个有序数组[LeetCode]

题目描述

思路

因为nums1保证空间足够，且nums1和nums2都是有序数组，可以考虑从后向前遍历两个数组，这样可以避免从前向后时nums1数组因为排序导致的交换问题。

代码实现1

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        // 2个数组一共m+n个元素，合并之后最大索引即为m+n-1
        int max_index = m + n - 1;
        int i = m -1, j = n - 1; // 从后向前遍历
        while(i >= 0 || j >= 0) {
            if (i >= 0 && j >= 0) {
                nums1[max_index--] = nums1[i] >= nums2[j] ? nums1[i--] : nums2[j--];
            } else if (i >= 0) {
                nums1[max_index--] = nums1[i--];
            } else {
                nums1[max_index--] = nums2[j--];
            }
        }
    }
};
复制代码

代码实现2

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int max_index = m + n - 1;
        m--; n--;
        // 这里不用或的条件，而用与，即先遍历结束其中一个数组
        // 如果是nums1先结束，则继续nums2，也就是后一个while循环
        // 如果是nums2先结束，nums1不用继续遍历，结果本身就是保存在nums1里，这就是最终结果了
        while(m >= 0 && n >= 0) {
            nums1[max_index--] = nums1[m] >= nums2[n] ? nums1[m--] : nums2[n--];
        }
        
        while(n >= 0) {
            nums1[max_index--] = nums2[n--];
        }
    }
};
复制代码