leetcode------合併排序數組

題目

合併排序的數組
給定兩個排序後的數組 A 和 B，其中 A 的末端有足夠的緩衝空間容納 B。 編寫一個方法，將 B 合併入 A 並排序。
初始化 A 和 B 的元素數量分別爲 m 和 n。
示例:
輸入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6], n = 3
輸出: [1,2,2,3,5,6]
說明:
A.length == n + m

第一次看到這道題，怎麼好像有點眼熟，原來是歸併排序的最後一步，合併兩個有序數組，採用雙指針的方式進行即可。

代碼

class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
      int[] tmp=new int[m+n];
      int i=0,j=0,k=0;
      while(i<m&&j<n){
          if(A[i]<B[j]){
              tmp[k++]=A[i++];
          }else{
              tmp[k++]=B[j++];
          }
      }
      while(i<m){
          tmp[k++]=A[i++];
      }
      while(j<n){
          tmp[k++]=B[j++];
      }
       for (int x = 0; x < m+n; x++){
            A[x] = tmp[x];
        }
    }
}

然後看題解 我擦勒，居然可以從後往前？？？，而且居然可以不需要額外的內存空間，太帥了。這裏忍不住mark下來

public void merge(int[] A, int m, int[] B, int n) {
        // 先確保將其中一個數組中的數字遍歷完
        while (m > 0 && n > 0) {
            // 對比選出較大的數放在 m + n - 1 的位置，並將選出此數的指針向前移動
            A[m + n - 1] = A[m - 1] > B[n - 1] ? A[m-- - 1] : B[n-- - 1];
        }
        // 剩下的數都比已經遍歷過的數小
        // 如果 m 不爲 0，則 A 沒遍歷完，都已經在 A 中不用再管
        // 如果 n 不爲 0，則 B 沒遍歷完，直接全移到 A 中相同的位置
        while (n > 0) {
            A[n - 1] = B[n - 1];
            n--;
        }
    }
作者：1ujin
鏈接：https://leetcode-cn.com/problems/sorted-merge-lcci/solution/5xing-dai-ma-shuang-bai-dao-xu-fa-shi-jian-fu-za-d/
來源：力扣（LeetCode）
著作權歸作者所有。商業轉載請聯繫作者獲得授權，非商業轉載請註明出處。