1. 程式人生 > >[leetcode]88.Merge Sorted Array

[leetcode]88.Merge Sorted Array

number class 是把 ali 題目 ++ into nts num

題目

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:

The number of elements initialized in nums1 and nums2 are m and n respectively.
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.
Example:

Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3

Output: [1,2,2,3,5,6]

解法一

思路

最先想到的是把num2直接復制到nums1後面,然後進行個排序即可。時間復雜度取決於排序算法。

代碼

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

解法二

思路

這種方法直接創建一個新的m+n的數組,然後比較num1和nums2的當前元素,將較小的值插入新數組,依次循環即可。但是這種方法違反了題目的本意,因為題目並不想讓創建新數組,所以此種方法不寫代碼了,了解即可。

解法三

思路

這種方法比較精妙,由於合並後的數組必然是m+n的,所以我們可以從後往前開始賦值,先將nums1和nums2的較大值放到m+n-1的位置上,然後依次向前推。如果循環結束,當n不為0但m為0時,num1剩下的元素就在相應的位置,不用處理。當m不為0但n為0,那就把nums2剩下的數依次復制過去。

代碼

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

[leetcode]88.Merge Sorted Array