【LeetCode】88. 合併兩個有序陣列

Curryxin 發表於 2021-08-22
LeetCode

88. 合併兩個有序陣列

知識點:陣列;排序;雙指標

題目描述

給你兩個按 非遞減順序 排列的整數陣列 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] 。

解法一:排序

=這道題很簡單,可以直接把num2合併到num1上,也就是將兩個陣列合並,然後去排序;

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        for(int i = m; i < m+n; i++){
            nums1[i] = nums2[i-m];
        }
        for(int i = 0; i < m+n; i++){
            for(int j = 0; j < m+n-i-1; j++){
                if(nums1[j] > nums1[j+1]){
                    int temp = nums1[j];
                    nums1[j] = nums1[j+1];
                    nums1[j+1] = temp;
                }
            }
        }
    }
}

解法二:雙指標

上面沒有用到一個條件:兩個陣列都是排序好的,所以可以定義兩個指標,對於兩個陣列都從頭到尾遍歷。經典的以空間換時間;

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int[] arr = new int[m+n];
        int index1 = 0, index2 = 0;
        int index = 0;
        while(index1 < m || index2 < n){
            if(index1 == m){
                arr[index++] = nums2[index2++];
            }else if(index2 == n){
                arr[index++] = nums1[index1++];
            }else if(nums1[index1] <= nums2[index2]){
                arr[index++] = nums1[index1++];
            }else if(nums1[index1] > nums2[index2]){
                arr[index++] = nums2[index2++];
            }
        }
        for(int i = 0; i < m+n; i++){
            nums1[i] = arr[i];
        }
    }
}