0
点赞
收藏
分享

微信扫一扫

JAVA练习189-合并排序的数组

丹柯yx 2022-03-11 阅读 75

给定两个排序后的数组 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

分析:

方法1:快速排序

最简单的方式,将 B 的数字复制到 A 的末端,然后对 A 进行快速排序。

时间复杂度:O((m+n)^log(m+n))         m,n 为两个数组的长度
空间复杂度:O(log(m+n))

class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
        //将B的值赋到A的末端
        for(int i = 0; i < n; ++i){
            A[m+i] = B[i];
        }
        //排序
        quickSort(A, 0, m + n - 1);
    }

    //快速排序
    public void quickSort(int[] arr, int start, int end){
        if(start >= end){
            return;
        }
        //定义指针
        int i = start, j = end, temp = arr[start];
        //遍历
        while(i < j){
            while(i < j && arr[j] >= temp){
                j--;
            }
            while(i < j && arr[i] <= temp){
                i++;
            }
            //交换
            int tt = arr[i];
            arr[i] = arr[j];
            arr[j] = tt;
        }
        //交换
        arr[start] = arr[i];
        arr[i] = temp;
        //继续对两边进行操作
        quickSort(arr, i+1, end);
        quickSort(arr, start, i-1);
    }
}

方法2:归并排序(双指针)

因为两个数组都是排好序的,因此我们可以利用这一点进行归并排序。定义双指针分别指向两个数组末位(A 指向有数的末位),每次比较筛选出最大值放在 A 的末位,直到遍历完成。A 的真实末位可以额外定义一个指针。

时间复杂度:O(m+n)         m,n 为两个数组的长度
空间复杂度:O(1)

class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
        //定义双指针
        int i = m-1, j = n-1, k = A.length - 1;
        //遍历
        while(i >= 0 && j >= 0){
            //A中元素比B大
            if(A[i] >= B[j]){
                A[k--] = A[i--];
            }
            else{
                A[k--] = B[j--];
            }
        }
        //复制没比较完的
        while(j >= 0){
            A[k--] = B[j--];
        }
    }
}

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sorted-merge-lcci

举报

相关推荐

0 条评论