給你兩個(gè)有序整數(shù)數(shù)組 nums1 和 nums2，請你將 nums2 合并到 nums1 中，使 nums1 成為一個(gè)有序數(shù)組

https://leetcode-cn.com/problems/merge-sorted-array/

初始化 nums1 和 nums2 的元素?cái)?shù)量分別為 m 和 n 。你可以假設(shè) 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]

示例2：

輸入：nums1 = [1], m = 1, nums2 = [], n = 0
輸出：[1]

提示：

nums1.length == m + n
nums2.length == n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[i] <= 109

Java解法

思路：

• 一道簡單題，不需要額外空間，直接在nums1上插入
• 分別記錄兩數(shù)組遍歷位置，插入進(jìn)行排序
• 一道簡單題做起來也出錯(cuò)兩次0.0
  • 注意輸入?yún)?shù)都是必要參數(shù)，我以為全是正數(shù)，所以以0判斷是不是到末尾了

package sj.shimmer.algorithm.m2;

import sj.shimmer.algorithm.Utils;

/**
 * Created by SJ on 2021/3/11.
 */

class D45 {
    public static void main(String[] args) {
//        int[] nums1 = {1, 2, 3, 0, 0, 0};
//        int[] nums2 = {2, 5, 6};
//        int[] nums1 = {2,0};
//        int[] nums2 = {1};

        int[] nums1 = {-1, 0, 0, 3, 3, 3, 0, 0, 0};
        int[] nums2 = {1, 2, 2};
        merge(nums1, 6, nums2, 3);
        Utils.logArray(nums1);
    }
    public static void merge(int[] nums1, int m, int[] nums2, int n) {
        int index1 = 0;
        int index2 = 0;
        int length = nums1.length;
        while (index1<length&&index2<n) {
            if (nums1[index1]<nums2[index2]) {
                if (index1>=m) {
                    nums1[index1]=nums2[index2];
                    m++;
                    index2++;
                }
                index1++;
            }else {
                //nums1后移
                for (int i = length -1; i >index1; i--) {
                    nums1[i]=nums1[i-1];
                }
                nums1[index1] =nums2[index2];
                m++;
                index2++;
                index1++;
            }
        }
    }
}

image

官方解

https://leetcode-cn.com/problems/merge-sorted-array/solution/he-bing-liang-ge-you-xu-shu-zu-by-leetcode/

1. 合并后排序

   最簡單直接的做法，時(shí)間復(fù)雜度差，但代碼簡單就兩行

   System.arraycopy(nums2, 0, nums1, m, n);
   Arrays.sort(nums1);
   

   • 時(shí)間復(fù)雜度：O((n+m)log(n+m))
   • 空間復(fù)雜度： O(1)

2. 雙指針

   從前往后：用額外的M空間存儲(chǔ)nums1的數(shù)據(jù)，雙指針比較存入數(shù)組中，得到最優(yōu)時(shí)間復(fù)雜度

   從后往前：不需要額外空間，只需要比較從末尾開始存儲(chǔ)即可，NICE

   public void merge(int[] nums1, int m, int[] nums2, int n) {
       // two get pointers for nums1 and nums2
       int p1 = m - 1;
       int p2 = n - 1;
       // set pointer for nums1
       int p = m + n - 1;
       // while there are still elements to compare
       while ((p1 >= 0) && (p2 >= 0))
           // compare two elements from nums1 and nums2 
           // and add the largest one in nums1 
           nums1[p--] = (nums1[p1] < nums2[p2]) ? nums2[p2--] : nums1[p1--];
   
       // add missing elements from nums2
       System.arraycopy(nums2, 0, nums1, 0, p2 + 1);
   }
   

   • 時(shí)間復(fù)雜度：O(n+m)
   • 空間復(fù)雜度： O(1)