合并两个有序数组

leetcode学习笔记

88. 合并两个有序数组

给你两个有序整数数组 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]

提示：

-10^9 <= nums1[i], nums2[i] <= 10^9
nums1.length == m + n
nums2.length == n

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/merge-sorted-array

思路

我们直接选择p1，p2作为两个数组的指针，p作为目标数组的指针，先把数组nums1复制到nums1_copy中，然后用循环比较nums1_copy和nums2的数的大小，较小的添加进nums1中，因为我们默认数组nums1的长度是m+n，不会产生数组越界的情况。当某一数组没有遍历完，我们再把剩下的数添加进nums1中。

代码

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
         int[] nums1_copy = new int[m];
        System.arraycopy(nums1,0,nums1_copy,0,m);
        //这个函数的各个参数分别是，来源数组、来源数组的起始位置、目标数组、目标数组的起始位置、目标数组的结束位置
        int p1=0,p2=0,p=0;//各个指针
        while(p1<m&&p2<n)
        {
            if(nums1_copy[p1]<nums2[p2])
            {
                nums1[p++]=nums1_copy[p1++];
                //可翻看++p和p++的区别
            }
            else{
                nums1[p++]=nums2[p2++];
            }
        }
        if(p1<m)System.arraycopy(nums1_copy,p1,nums1,p1+p2,m+n-p1-p2);
        //未遍历完的nums1_copy数组添加入nums1
        if(p2<n)System.arraycopy(nums2,p2,nums1,p1+p2,m+n-p1-p2);
        //未遍历完的nums2数组
        System.out.print(Arrays.toString(nums1));
       //以字符串的形式输出
    }
}