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)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章