leetcode刷题记录--合并两个有序数组(双指针法)

题目描述

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

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

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

示例一：

输入：nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出：[1,2,2,3,5,6]
解释：需要合并 [1,2,3] 和 [2,5,6] 。
合并结果是 [1,2,2,3,5,6] ，其中斜体加粗标注的为 nums1 中的元素。

示例二：

输入：nums1 = [1], m = 1, nums2 = [], n = 0
输出：[1]
解释：需要合并 [1] 和 [] 。
合并结果是 [1] 。

示例三：

输入：nums1 = [0], m = 0, nums2 = [1], n = 1
输出：[1]
解释：需要合并的数组是 [] 和 [1] 。
合并结果是 [1] 。
注意，因为 m = 0 ，所以 nums1 中没有元素。nums1 中仅存的 0 仅仅是为了确保合并结果可以顺利存放到 nums1 中。

解法一：暴力合并后排序

此方法就是把nums2加到nums1的末尾，对新得到的nums1进行排序
代码如下：

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
       for(int i =0;i<n;i++){
           nums1[m+i]=nums2[i];
       }
       Arrays.sort(nums1);
    }
}

解法二：双指针法

思路

上述的直接添加到末尾后进行排序，显然并没有利用到题目里给的nums1和nums2都已经排好序这一条件，所以我们使用双指针法进行改善

1.定义两个指针，分别指向nums1和nums2的首元素，定义一个目标数组用来存放我们排序的结果
2.取出俩个数组首个元素，把较小的那个赋值到目标数组中
3.把目标数组赋值给nums1

代码实现

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
       //指向nums1的第一个元素
       int p1 = 0;
       //指向nums2的第一个元素
       int p2 = 0;
       //储存需要添加进新数组的元素
       int add = 0;
       //储存我们需要的数组
       int[] sort = new int[m+n];

       while(p1<m||p2<n){
           //前面两个if用来判断两个数组是否有一个已经全部添加
           if (p1==m){
               add = nums2[p2];
               p2++;
           }else if(p2==n){
               add = nums1[p1];
               p1++;
           }else if (nums1[p1]<nums2[p2]){
               add = nums1[p1];
               p1++;
           }else{
               add = nums2[p2];
               p2++;
           }
           //p1+p2就是我们一共移动了几位，减去1就是要添加的位置
           sort[p1+p2-1]=add;
       }

       //最后赋值给nums1
       for(int i=0;i<m+n;i++){
           nums1[i]=sort[i];
       }
    }
}