Leetcode 第88题 合并两个有序数组题解

题目描述

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

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m +n，这样它就有足够的空间保存来自 nums2 的元素。

示例 1：

输入：nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出：[1,2,2,3,5,6]

来源：力扣（LeetCode）
链接：LeetCode

题目分析

（1）不考虑空间和时间的情况较为简单，直接把nums2的元素放入到nums1数组后面，然后调用排序Arrays.sort()即可。

（2）但是一般情况下不考虑此解；此处用双指针法解，在不考虑空间复杂度的情况下，可开一个m+n大小的数组，用于暂存结果，两指针分别指向两数组的第一个元素，然后比较这两个元素，将较小的元素加入到新数组，并把指针指向下一个元素，以此遍历直至一个数组为空，然后把另外一个数组所有元素加入新数组即可。

（3）如果考虑空间，则可以利用nums1数组后半部分空闲空间，同样用指针法（从后往前）即可，此处不详细写此解法。

代码

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int p=0;
        int q=0;
        int cur=0;
        int []res = new int[m+n];
        while(p<m||q<n){
            if(p>=m){
                res[cur++]=nums2[q++];
            }
            else if(q>=n){
                res[cur++]=nums1[p++];
            }
            else if(nums1[p]>=nums2[q]){
                res[cur++]=nums2[q++];
            }
            else
                res[cur++]=nums1[p++];
        }
        for(int i=0;i<m+n;i++){
            nums1[i]=res[i];
        }
    }
}