88. 合并两个有序数组

问题描述

• 给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。
• 请你 合并 nums2 到 nums1 中，使合并后的数组同样按 非递减顺序 排列。
• 注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为n。

示例1：
输入：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 中的元素。

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

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

题目来源：LeetCode官方
链接地址：https://leetcode.cn/problems/merge-sorted-array/

思路

• 注意读题，nums1的长度为m+n，但一开始时只有前m个元素是有实际意义的，后面n个位置全部填充的0，nums2的长度为n，且两个数组和返回的数组都要按非递减排序，其实也就是升序，这些就是本题的关键信息了
• 两个有序数组进行合并，而且最后的返回结果全部放在第一个数组里，我们很容易想到，只要对两个数组进行遍历，然后只要nums2的元素找到在nums1中合适的位置，将其插入进去就好，思路很简单，但仔细一想，你会发现，这种方法会导致元素后移的操作非常频繁，从而花费大量的时间，所以在这里我们不予考虑
• 上面的思路是从前往后的，从题目给的信息我们注意到nums1的长度是m+n，也就是说长度是够的，并且恰好是能填满的，那我们为什么不考虑一下从后往前的思路呢?我们令两个变量分别指向nums1的第m个位置，nums2的第n个位置，比较二者所指的元素，较大的那个直接放到nums1数组的末尾，然后两个变量相应进行前移，再进行比较，以此类推。这样我们就不用考虑移位操作，只要进行元素覆盖就行了。

核心代码

    public static void merge_tail(int[] nums1, int m, int[] nums2, int n) {
        int p = m - 1;//p指向nums1非0元素的最后一个
        int q = n - 1;//q指向nums2的最后一个元素
        int tail = m + n - 1;//tail指向nums1数组最后一个元素
        //前提是nums1，nums2都有序
        while (p >= 0 || q >= 0) {//只要两个数组其中有一个没空，就循环
            if (p == -1) {//nums1已经空了，但nums2没空,将nums2依次加入到nums1中
                nums1[tail--] = nums2[q--];
            } else if (q == -1) {//nums2已经空了，但nums1没空
                nums1[tail--] = nums1[p--];
            } else if (nums1[p] > nums2[q]) {//nums1的元素大
                nums1[tail--] = nums1[p--];
            } else {//这个情况就是nums2的元素大了
                nums1[tail--] = nums2[q--];
            }
        }
    }

代码解释

直接见代码里的注释

运行效果

完整源代码

源代码这里还提供了另外一种思路：就是直接将两个数组拼接在一起，然后进行排序就行

package com.easy;

public class array_merge {
    public static void main(String[] args) {
        int[] arr1 = {1, 2, 3, 0, 0, 0};
        int[] arr2 = {4, 9, 8};
        merge(arr1, 3, arr2, 3);
        for (int val : arr1) {
            System.out.print(val);
        }
    }

    /**
     * 解法1：这种优化不是很好，时空复杂度有点高
     * 思路：现将两个数组拼接起来，然后利用排序算法解决
     *
     * @param nums1
     * @param m
     * @param nums2
     * @param n
     */
    public static void merge(int[] nums1, int m, int[] nums2, int n) {
        //m=0的话，需要把nums2里的元素移到nums1里，如果n=0的话，就不需要合并了
        if (m == 0) {
            for (int i = 0; i < n; i++) {
                nums1[i] = nums2[i];
            }
        }
        //先将两个数组拼接在一起，然后排序
        for (int i = m, j = 0; i < m + n && j < n; i++, j++) {
            nums1[i] = nums2[j];
        }
        Bubble(nums1);
    }

    /**
     * 解法2：时空复杂度低，优化后的版本
     * 思路：注意到nums1后面的元素为0，也就是说我们可以没有顾虑地覆盖掉它，这就要求我们每次选出两数组中最大的元素依次从后往前填入
     *
     * @param nums1
     * @param m
     * @param nums2
     * @param n
     */
    public static void merge_tail(int[] nums1, int m, int[] nums2, int n) {
        int p = m - 1;//p指向nums1非0元素的最后一个
        int q = n - 1;//q指向nums2的最后一个元素
        int tail = m + n - 1;//tail指向nums1数组最后一个元素
        //前提是nums1，nums2都有序
        while (p >= 0 || q >= 0) {//只要两个数组其中有一个没空，就循环
            if (p == -1) {//nums1已经空了，但nums2没空,将nums2依次加入到nums1中
                nums1[tail--] = nums2[q--];
            } else if (q == -1) {//nums2已经空了，但nums1没空
                nums1[tail--] = nums1[p--];
            } else if (nums1[p] > nums2[q]) {//nums1的元素大
                nums1[tail--] = nums1[p--];
            } else {//这个情况就是nums2的元素大了
                nums1[tail--] = nums2[q--];
            }
        }
    }


    /**
     * 冒泡排序，升序
     * 思想为比较相邻的两元素，将较大的元素不断往后移，一次排序的结果就是将最大的元素推到最后的位置，进行多次排序即可
     *
     * @param nums
     */
    public static void Bubble(int[] nums) {
        for (int i = 0; i < nums.length - 1; i++) {//i控制排序次数，因为每一次排序只是将一个元素推到最终位置，所以要进行多次，前n-1次排序完成后，第n个元素自然有序
            for (int j = 0; j < nums.length - i - 1; j++) {//j用于比较相邻元素，并交换位置，对于最后部分已经排好的元素就不用去比较了，故此n-1-i
                if (nums[j] > nums[j + 1]) {//升序排序，只在前一个元素比后一个元素大的时候交换位置
                    int temp = nums[j];
                    nums[j] = nums[j + 1];
                    nums[j + 1] = temp;
                }
            }
        }
    }

    /**
     * 选择排序，升序
     * 第一次遍历n-1个数，找到最小的数值与第一个元素交换
     * 第二次遍历n-2个数，找到最小的数值与第二个元素交换
     * 依次类推，直到排序完成
     *
     * @param nums
     */
    public static void Select(int[] nums) {
        for (int i = 0; i < nums.length - 1; i++) {//同样的，前n-1次排序完成后，最后一个自然有序，控制排序次数
            int min = i;//初始化为待插入位置的元素的下标
            //遍历寻找该趟次下的最小元素
            for (int j = i + 1; j < nums.length; j++) {//控制元素，假设进行m次排序后，那么数组就分为了前m部分（有序），后n-m部分（无序），有序部分不需要再排序，所以j = i+1
                if (nums[min] > nums[j]) {
                    min = j;
                }
            }
            //循环结束后，找到最小元素，将nums[i]元nums[min]进行交换
            int temp = nums[i];
            nums[i] = nums[min];
            nums[min] = temp;
        }
    }

}

运行截图

这里就直接展示先拼接，再排序的截图，merge_tail方法的截图不展示了，效果是一样的