面試題：合併2個有序數組（leetcode88）

給定兩個有序整數數組 nums1 和 nums2，將 nums2 合併到 nums1 中，使得 num1 成爲一個有序數組。

說明:

• 初始化 nums1 和 nums2 的元素數量分別爲 m 和 n。
• 你能夠假設 nums1 有足夠的空間（空間大小大於或等於 m + n）來保存 nums2 中的元素。

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

輸出: [1,2,2,3,5,6]

我本身寫了一遍，時間複雜度是O(m+n)

package javaDemo;

public class ArrDemo {
    
    public static void main(String[] args) {
        int[] nums1 =  new int[] {1,2,3};
        int[] nums2 =  new int[] {2,5,6};
        
        
        for (int num : merge(nums1,3,nums2,3)) {
            System.out.println(num);
        }
            

    }
    
    public static int[] merge(int[] nums1, int m, int[] nums2, int n) {
        int[] k =new int[m+n];
        int i=0,j=0,kindex=0;
        while (i<m && j<n && kindex<m+n){
            if (nums1[i] <= nums2[j]){
                k[kindex++] = nums1[i];
                i++;                    
            }
            else{
                k[kindex++] = nums2[j];
                j++;
            }
            
            //說明nums1已經結束遍歷，把nums2的剩餘全給k
            if (i==m) {
                while(j<n) {
                    k[kindex++] = nums2[j++];  
                }
            }

            if (j==n) {
                while(i<m) {
                    k[kindex++] = nums1[i++];  
                }
            }
        }
        
        nums1 = k;
        return  nums1;
        
    }

}

還有一個更高效的:

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