【Leetcode常见面试题解析—合并两个有序数组】

题目描述

给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。

请你 合并 nums2 到 nums1 中，使合并后的数组同样按 非递减顺序 排列。

注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。

解题思路

首先这道题有个关键条件：两个子数组是有序的
那么我们可以很自然的想到采用【双指针】的方法，针对两个数组对应位置的元素进行比较，找到较小/较大者进行排序，故而解题思路有以下两种：

1. 较简单版: 创建一个长度为m+n的新数组，两个指针各自从数组的首元素向后找，找到每个位置上的最小值放入新数组对应的位置里

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        vector<int>res(m+n);
        int p=0;//p指向数组nums1
        int q=0;//q指向数组nums2
        int cur=0;//cur指向新数组
        while(p<m&&q<n)//两个数组都没走完
        {
            if(nums1[p]<=nums2[q])
            {
                res[cur]=nums1[p];
                cur++;
                p++;
            }
            else
            {
                res[cur]=nums2[q];
                cur++;
                q++;
            }
        }
        while(p<m)//q走完了 p还没走完
        {
            res[cur]=nums1[p];
            cur++;
            p++;
        }
        while(q<n)//p走完了 q还没走完
        {
            res[cur]=nums2[q];
            cur++;
            q++;
        }
      nums1=res;//将res赋值给nums1
    }
};

2. 较难版本: 不新建数组，采用交换元素的方式在nums1上直接操作，这个时候需要再借助一个指针，指向长数组的最后一个元素

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int i=m+n-1;//i指向长数组的最后一个元素
        int p=m-1;//p指向nums1最后一个非0元素
        int q=n-1;//q指向nums2最后一个非0元素
        while(q>=0)
        {
            if(p>=0&&nums1[p]<=nums2[q])//p对应的值不大于q对应的值时
            {
              nums1[i]=nums2[q];
              q--;
              i--;
            }
            else if(p>=0&&nums1[p]>nums2[q]){//p对应的值大于q对应的值时
                nums1[i]=nums1[p];
                p--;
                i--;
            }
            else{//这种情况是p越界了，对应nums1数组首元素
                nums1[i]=nums2[q];
                i--;
                q--;
            }
        }
    }
};