【leetcode刷題】4.有序數組合并——Java版

前言

這是一條美團一面遇到的題目，屬于簡單級别。

仔細思考和昨天的

滑動視窗

有異曲同工之妙！

Question

難度：簡單

給你兩個有序整數數組 nums1 和 nums2，請你将 nums2 合并到 nums1 中，使 nums1 成為一個有序數組。

初始化 nums1 和 nums2 的元素數量分别為 m 和 n 。你可以假設 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      

Solution

• 定義兩個指針分别指向

  nums1

  和

  nums2

  的元素。
• 定義一個新數組的指針
• 注意：

  nums1.length == m + n

  ，這個資訊告訴我們不需建立數組，

  原地修改

  即可。
• 為了不破壞原有數組，從後向前周遊，邊比較邊修改。

Code

/**
 * @author yitiaoIT
 */
class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int indexM1=m-1;
        int indexN2=n-1;
        int index=m+n-1;
        while (indexN2>=0||indexM1>=0){
            if (indexM1<=-1){
                nums1[index--]=nums2[indexN2--];
            }else if(indexN2<=-1){
                nums1[index--]=nums1[indexM1--];
            }else{
                nums1[index--]=nums1[indexM1]>nums2[indexN2]?nums1[indexM1--]:nums2[indexN2--];
            }

        }
    }
}      

Result

複雜度分析

• 時間複雜度：O(max(m,n))

我是一條，一個在網際網路摸爬滾打的程式員。

道阻且長，行則将至。大家的 【點贊，收藏，關注】 就是一條創作的最大動力，我們下期見！