LeetCode精选TOP面试题88.合并两个有序数组

题目描述

  • 给定两个 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数m和n,分别表示nums1和nums2中的元素数目。
  • 合并nums2到nums1中,使合并后的数组同样按 非递减顺序 排列。最终合并后数组不应由函数返回,而是存储在数组 nums1 中。
  • 为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。
  • nums1.length == m + n;
  • nums2.length == n
  • 0 <= m, n <= 200
  • 1 <= m + n <= 200
  • -109 <= nums1[i], nums2[j] <= 109
example
input  : nums1 = {
    
    1,2,3,0,0,0}, m = 3, nums2 = {
    
    2,5,6}, n = 3
output : {
    
    1,2,2,3,5,6}
input  : nums1 = {
    
    1}, m = 1, nums2 = {
    
    }, n = 0
output : {
    
    1}
input  : nums1 = {
    
    0}, m = 0, nums2 = {
    
    2}, n = 1
output : {
    
    2}

解题思路

关键点:两个数组都是有序数组

思路1 逆序双指针

  • 因为nums1的后几位是与nums2等长的0元素,所以逆序用双指针分别指向两个数组末尾,找到最大元素,将其值赋给nums1尾部,依次循环向前
  • 当某个指针遍历结束时停止第一轮遍历,此时肯定有一个数组已经被完全纳入到nums1尾部
  • 且此时并不会对nums1和nums2这两个数组的未排序元素造成影响
  • 之后,若nums1未被全部遍历,则不需要继续处理,因为此时数组必定已经有序;若nums2未被全部遍历,则需继续逆序遍历nums2,将其元素值依次赋值到nums1中

思路2 合并数组后排序

代码(Java)

思路1代码

public class Solution1 {
    
    
    public void merge(int[] nums1, int m, int[] nums2, int n) {
    
    
        if (n == 0) {
    
    
            return;
        }
        int length = m + n;
        int i = m - 1, j = n - 1;
        while (i >= 0 && j >= 0) {
    
    
            nums1[--length] = nums1[i] > nums2[j] ? nums1[i--] : nums2[j--];
        }
        while (i >= 0) {
    
    
            nums1[--length] = nums1[i--];
        }
        while (j >= 0) {
    
    
            nums1[--length] = nums2[j--];
        }
    }
}

思路2代码

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

猜你喜欢

转载自blog.csdn.net/flower_48237/article/details/120959355