【算法系列篇】分治-归并

简介: 【算法系列篇】分治-归并

前言

上一篇算法文章,我们介绍了分治-快排的算法,今天我将为大家分享关于分治的另外一种算法——归并。

什么是归并算法

归并算法是一种常用的排序算法,它采用分治策略将待排序的数组分解为更小的子数组,然后逐步合并这些子数组以获得最终的有序数组。归并排序的主要思想是将两个有序的子数组合并成一个有序的数组。


归并算法通常包含以下步骤:


分解(Divide):将待排序的数组递归地分解为规模更小的子数组,直到每个子数组只有一个元素或为空。


解决(Conquer):通过递归地排序子数组,将其转化为有序的子数组。这通常是通过继续将子数组进一步分解并排序的方式实现的。


合并(Merge):将两个有序的子数组合并成一个有序的数组。该步骤的实现方式是比较两个子数组的元素,并按照顺序合并到一个新的数组中,直到所有元素都被合并。


归并排序的时间复杂度是O(nlogn),其中n是待排序数组的长度。它的主要优点包括:


稳定性:归并排序是一种稳定的排序算法,即相等元素的相对顺序不会被改变。


适用性:归并排序适用于各种数据结构,尤其在外部排序中,它对于大规模数据的排序效果明显。


然而,归并排序也存在一些缺点:


额外空间消耗:归并排序需要额外的空间来存储临时的子数组和合并结果,这可能对内存消耗造成一定影响。


递归调用:归并排序的实现通常使用递归调用,对于大规模数据的排序可能导致递归深度增加,从而增加了额外的函数调用开销。


总结而言,归并排序是一种高效、稳定的排序算法,通过分治策略将待排序的数组分解为更小的子数组,然后逐步合并这些子数组以获得最终的有序数组。尽管归并排序需要额外的空间和函数调用开销,但它在实践中被广泛使用,特别适用于对大规模数据进行排序。

1. 排序数组

https://leetcode.cn/problems/sort-an-array/

1.1 题目要求

给你一个整数数组 nums,请你将该数组升序排列。


示例 1:

输入:nums = [5,2,3,1]
输出:[1,2,3,5]

示例 2:

输入:nums = [5,1,1,2,0,0]
输出:[0,0,1,1,2,5]


提示:

  • 1 <= nums.length <= 5 * 104
  • -5 * 104 <= nums[i] <= 5 * 104
class Solution {
    public int[] sortArray(int[] nums) {
    }
}


1.2 做题思路

不知道大家是否做过将两个有序数组合并为一个有序数组,我们的归并算法就是通过将两个数组合并为一个有序数组来实现的。而归并的思想就是就一整个数组从中间将数组分为两个数组,然后再继续将这两个数组分别从中间分开,直到将这两部分的数组分为只有一个元素的两部分数组,然后将这两个数组通过合并两个数组的操作来进行合并,合并完成之后的数组就成为了一个有序的数组,然后继续将这两个有序的数组通过合并数组的操作继续进行合并,直到将这些数组合并为一个最大的数组。


1.3 Java代码实现

class Solution {
    //因为每一次递归都需要创建临时的数组来存储两个数组排序之后的结果,
    //每次都向申请内存,速度会很慢
    //所以我们直接申请一个跟nums同样大小的数组
    int[] tmp;
    public int[] sortArray(int[] nums) {
        int n = nums.length;
        tmp = new int[n];
        mergeSort(nums,0,n -1);
        return nums;
    }
    private void mergeSort(int[] nums, int left, int right) {
        //当数组中只有一个元素或者区间不成立的时候,结束递归
        if(left >= right) return;
        int mid = left + (right - left) / 2;
        //先排序mid左右两边的数组,最后在将左右两边的有序数组进行合并
        mergeSort(nums,left,mid);
        mergeSort(nums,mid + 1,right);
        int cur1 = left,cur2 = mid + 1,i = 0;
        while(cur1 <= mid && cur2 <= right) {
            tmp[i++] = nums[cur1] < nums[cur2] ? nums[cur1++] : nums[cur2++];
        }
        //处理没到达数组结尾的数组
        while(cur1 <= mid) tmp[i++] = nums[cur1++];
        while(cur2 <= right) tmp[i++] = nums[cur2++];
        //将临时排序之后数组的结果更新到我们原本的数组中
        for(int j = left; j <= right; j++) nums[j] = tmp[j - left];
    }
}

2. 数组中逆序对

https://leetcode.cn/problems/shu-zu-zhong-de-ni-xu-dui-lcof/description/

2.1 题目要求


在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数。

示例 1:

输入: [7,5,6,4]
输出: 5

限制:

  • 0 <= 数组长度 <= 50000
class Solution {
    public int reversePairs(int[] nums) {
    }
}

2.2 做题思路

我们先来看看如何使用暴力解法来解决这个问题:用两层循环来遍历数组,i 从 0 开始,j 则从 i 的下一个位置开始,看 j 所指向的位置是否小于 i 所指的位置,如果是则逆序对总数加一。这样虽然简单,但是时间复杂度达到了 O(N^2),是跑不过去的,那么我们就需要对暴力解法进行优化,问题就在于我们该如何优化呢?


这样想,如果我们将数组分为两个部分,先计算左边部分的数组中的所有逆序对,然后计算右边部分数组中的所有逆序对,最后在左边数组一次选择一个数字,看右边数组中是否元素能与左边数组中的那个元素构成逆序对,这样的思路其实跟暴力解法是相同的,时间复杂度都是 O(N^2),那么当我们分别在左边数组中和右边数组中找到了符合的逆序对的话,我们是否可以将这两部分数组进行排序呢?当进行了排序之后,再分别在左边数组中依次拿元素与右边数组中的元素进行比较,假设我们按照升序的方式进行排序,当左边部分遇到比右边数组部分大的元素的话,那么左边数组部分从这个元素开始到左边部分数组结束的位置是否都大于右边数组的该元素呢?


如果nums[cur1] < nums[cur2] 则继续让 cur1 向右移动,直到出现逆序对的情况,这样 cur1 和 cur2 指针都不用返回,就较少了很多的重复比较,使得时间复杂度提升为

O(N*logN)。

2.3 Java代码实现

(1)升序方法

class Solution {
    int[] tmp;
    public int reversePairs(int[] nums) {
        int n = nums.length;
        tmp = new int[n];
        return mergeSort(nums,0,n-1);
    }
    private int mergeSort(int[] nums, int left, int right) {
        if(left >= right) return 0;
        int mid = left + (right - left) / 2;
        //统计逆序对的数量
        int ret = 0;
        //计算左边数组中逆序对的数量
        ret += mergeSort(nums,left,mid);
        //计算右边数组中逆序对的数量
        ret += mergeSort(nums,mid + 1,right);
        int cur1 = left, cur2 = mid + 1,i = 0;
        while(cur1 <= mid && cur2 <= right) {
            //当nums[cur1] <= nums[cur2] 的时候,只需要进行合并数组的操作
            if(nums[cur1] <= nums[cur2]) tmp[i++] = nums[cur1++];
            //当nums[cur1] > nums[cur2] 的时候,需要更新逆序对的数量,同时也需要合并数组
            else {
                ret += mid - cur1 + 1;
                tmp[i++] = nums[cur2++];
            }
        }
        while(cur1 <= mid) tmp[i++] = nums[cur1++];
        while(cur2 <= right) tmp[i++] = nums[cur2++];
        for(int j = left; j <= right; j++) nums[j] = tmp[j - left];
        return ret;
    }
}

(2)降序方法


我们也可以用降序排列的方式来实现归并排序,只是当使用降序的时候,需要在右边数组中统计逆序对的数量,因为当遇到 nums[cur1] > nums[cur2] 的时候,如果还是统计左边数组 mid - cur1 + 1 作为逆序对的数量的话,那么当 cur2 向后移动的时候,再遇到 nums[cur1] > nums[cur2] 的时候就会出现重复的情况,所以当我们以降序的方式排序的时候需要在右边数组中统计逆序对的数量。

class Solution {
    int[] tmp;
    public int reversePairs(int[] nums) {
        int n = nums.length;
        tmp = new int[n];
        return mergeSort(nums,0,n-1);
    }
    private int mergeSort(int[] nums, int left, int right) {
        if(left >= right) return 0;
        int mid = left + (right - left) / 2;
        //统计逆序对的数量
        int ret = 0;
        //计算左边数组中逆序对的数量
        ret += mergeSort(nums,left,mid);
        //计算右边数组中逆序对的数量
        ret += mergeSort(nums,mid + 1,right);
        int cur1 = left, cur2 = mid + 1,i = 0;
        while(cur1 <= mid && cur2 <= right) {
            //当nums[cur1] <= nums[cur2] 的时候,只需要进行合并数组的操作
            if(nums[cur1] <= nums[cur2]) tmp[i++] = nums[cur2++];
            //当nums[cur1] > nums[cur2] 的时候,需要更新逆序对的数量,同时也需要合并数组
            else {
                ret += right - cur2 + 1;
                tmp[i++] = nums[cur1++];
            }
        }
        while(cur1 <= mid) tmp[i++] = nums[cur1++];
        while(cur2 <= right) tmp[i++] = nums[cur2++];
        for(int j = left; j <= right; j++) nums[j] = tmp[j - left];
        return ret;
    }
}

3. 计算右侧小于当前元素的个数

https://leetcode.cn/problems/count-of-smaller-numbers-after-self/description/

3.1 题目要求

给你一个整数数组 nums ,按要求返回一个新数组 counts 。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。

示例 1:

输入:nums = [5,2,6,1]
输出:[2,1,1,0] 
解释:
5 的右侧有 2 个更小的元素 (2 和 1)
2 的右侧仅有 1 个更小的元素 (1)
6 的右侧有 1 个更小的元素 (1)
1 的右侧有 0 个更小的元素

示例 2:

输入:nums = [-1]
输出:[0]

示例 3:

输入:nums = [-1,-1]
输出:[0,0]

提示:

  • 1 <= nums.length <= 105
  • -104 <= nums[i] <= 104
class Solution {
    public List<Integer> countSmaller(int[] nums) {
    }
}

3.2 做题思路

这道题目跟上面的找逆序对的数量是类似的,逆序对是找前面的元素大于后面元素的个数,而这道题是需要找到数组中每个元素的之后的小于该元素的个数,也就是说这个题目返回的是多个数,而不是一个数。那么我们在使用分冶算法的过程中该如何记住对应数字的下标呢?因为在分冶的过程中加上排序,数组中元素的位置是不断变化的,但是这道题目按顺序返回数组中对用位置右边部分大小小于该位置元素的数量,所以这道题目如何记住原本数组对应元素的下标是很重要的。


那么我们如何记住原本数组元素的对应下标呢?我们可以创建一个 index 数组,当数组进行合并排序的时候,相应的 index 数组也跟着变化,这样就能使排序后数组对应位置的 index 数组中存放的是原本数组的下标。


3.3 Java代码实现

class Solution {
    int[] ret; //该数组中存放右侧小于当前元素的个数
    int[] index; //存放对应元素的下标
    int[] tmpNums; //排序后的临时数组
    int[] tmpIndex; //排序后对应元素的原本下标
    public List<Integer> countSmaller(int[] nums) {
        int n = nums.length;
        ret = new int[n];
        index = new int[n];
        //index 数组初始化
        for(int i = 0; i < n; i++) index[i] = i;
        tmpNums = new int[n];
        tmpIndex = new int[n];
        mergerSort(nums,0,n-1);
        List<Integer> list = new ArrayList<>();
        for(int x : ret) list.add(x);
        return list;
    }
    private void mergerSort(int[] nums, int left, int right) {
        if(left >= right) return;
        int mid = left + (right - left) / 2;
        mergerSort(nums,left,mid);
        mergerSort(nums,mid + 1,right);
        int cur1 = left, cur2 = mid + 1, i = 0;
        //降序
        while(cur1 <= mid && cur2 <= right) {
            if(nums[cur1] <= nums[cur2]) {
                tmpNums[i] = nums[cur2];
                tmpIndex[i++] = index[cur2++];
            }else {
            //因为前面的递归中可能出现了右侧小于当前位置的数,所以需要使用+=
                ret[index[cur1]] += right - cur2 + 1;
                tmpNums[i] = nums[cur1];
                tmpIndex[i++] = index[cur1++];
            }
        }
        while(cur1 <= mid) {
            tmpNums[i] = nums[cur1];
            tmpIndex[i++] = index[cur1++];
        }
        while(cur2 <= right) {
            tmpNums[i] = nums[cur2];
            tmpIndex[i++] = index[cur2++];
        }
        for(int j = left; j <= right; j++) {
            nums[j] = tmpNums[j - left];
            index[j] = tmpIndex[j - left];
        }
    } 
}

4. 翻转对

https://leetcode.cn/problems/reverse-pairs/description/

4.1 题目要求

给定一个数组 nums ,如果 i < j 且 nums[i] > 2*nums[j] 我们就将 (i, j) 称作一个重要翻转对。

你需要返回给定数组中的重要翻转对的数量。


示例 1:

输入: [1,3,2,3,1]
输出: 2

示例 2:

输入: [2,4,3,5,1]
输出: 3


注意:

  • 给定数组的长度不会超过50000。
  • 输入数组中的所有数字都在32位整数的表示范围内。
class Solution {
    public int reversePairs(int[] nums) {
    }
}

4.2 做题思路

这个题目跟前面的的两个题目都是类似的,只是这里不能在我们更新翻转对的时候就进行排序,因为翻转对的判断条件是 i < j 且 nums[i] > 2*nums[j],根据这个判断条件,我们不能判断出 nums[i] 和 nums[j] 哪个大,所以只能在更新完翻转对之后进行归并排序,其他的步骤基本上是类似的。

4.3 Java代码实现

(1)升序

class Solution {
    int[] tmp;
    public int reversePairs(int[] nums) {
        int n = nums.length;
        tmp = new int[n];
        return mergeSort(nums,0,n-1);
    }
    private int mergeSort(int[] nums, int left, int right) {
        if(left >= right) return 0;
        int ret = 0;
        int mid = left + (right - left) / 2;
        //统计左右两部分数组中翻转对的数量
        ret += mergeSort(nums,left,mid);
        ret += mergeSort(nums,mid + 1,right);
        int cur1 = left, cur2 = mid + 1, i = 0;
        while(cur2 <= right) {
        //这里需要使用nums[cur1] / 2.0 而不是nums[cur2] * 2,
        //因为可能会导致溢出
            while(cur1 <= mid && nums[cur1] / 2.0 <= nums[cur2]) cur1++;
            if(cur1 > mid) break;
            ret += mid - cur1 + 1;
            cur2++;
        }
    //升序
        cur1 = left;
        cur2 = mid + 1;
        while(cur1 <= mid && cur2 <= right) {
            tmp[i++] = nums[cur1] < nums[cur2] ? nums[cur1++] : nums[cur2++];
        }
        while(cur1 <= mid) tmp[i++] = nums[cur1++];
        while(cur2 <= right) tmp[i++] = nums[cur2++];
        for(int j = left; j <= right; j++) nums[j] = tmp[j - left];
        return ret;
    }
}

(2)降序

class Solution {
    int[] tmp;
    public int reversePairs(int[] nums) {
        int n = nums.length;
        tmp = new int[n];
        return mergeSort(nums,0,n-1);
    }
    private int mergeSort(int[] nums, int left, int right) {
        if(left >= right) return 0;
        int ret = 0;
        int mid = left + (right - left) / 2;
        ret += mergeSort(nums,left,mid);
        ret += mergeSort(nums,mid + 1,right);
        int cur1 = left, cur2 = mid + 1, i = 0;
        //降序
        while(cur1 <= mid) {
        //这里cur1不动,因为cur1向后移动只会越来越小,所以让cur2向后移动
            while(cur2 <= right && nums[cur1] / 2.0 <= nums[cur2]) cur2++;
            if(cur2 > right) break;
            ret += right - cur2 + 1;
            cur1++;
        }
        cur1 = left;
        cur2 = mid + 1;
        while(cur1 <= mid && cur2 <= right) {
            tmp[i++] = nums[cur1] <= nums[cur2] ? nums[cur2++] : nums[cur1++];
        }
        while(cur1 <= mid) tmp[i++] = nums[cur1++];
        while(cur2 <= right) tmp[i++] = nums[cur2++];
        for(int j = left; j <= right; j++) nums[j] = tmp[j - left];
        return ret;
    }
}

总结

归并排序是一种高效而稳定的排序算法,它利用了分治策略,将待排序的数组分解为更小的子数组,并逐步合并这些子数组以获得最终的有序数组。


归并排序算法的核心思想是将待排序数组递归地分解为规模更小、有序的子数组,然后通过合并操作将这些子数组有序地合并成一个大的有序数组。这种分解和合并的过程直到最终合并成排序后的完整数组。


归并排序算法具有以下优点:


稳定性:归并排序是一种稳定的排序算法,即相等元素的相对顺序不会被改变。这使得它特别适用于对具有多关键字排序要求的情况。


高效性:归并排序的时间复杂度为O(nlogn),其中n是待排序数组的长度。它具有较好的性能表现,并适用于大规模数据的排序。


此外,归并排序算法还具有一定的弹性和灵活性。它可以通过优化合并操作的实现方式,减少额外空间的消耗。此外,归并排序也适用于外部排序,可以处理存储在外部存储介质中的大规模数据。


尽管归并排序算法需要额外的空间和函数调用开销,但由于其稳定性和较好的时间复杂度,它在实际应用中被广泛采用。

相关文章
|
5月前
|
算法 开发者 Python
惊呆了!Python算法设计与分析,分治法、贪心、动态规划...这些你都会了吗?不会?那还不快来学!
【7月更文挑战第10天】探索编程巅峰,算法至关重要。Python以其易读性成为学习算法的首选。分治法,如归并排序,将大问题拆解;贪心算法,如找零问题,每步求局部最优;动态规划,如斐波那契数列,利用子问题解。通过示例代码,理解并掌握这些算法,提升编程技能,面对挑战更加从容。动手实践,体验算法的神奇力量吧!
77 8
|
1月前
|
算法 Python
在Python编程中,分治法、贪心算法和动态规划是三种重要的算法。分治法通过将大问题分解为小问题,递归解决后合并结果
在Python编程中,分治法、贪心算法和动态规划是三种重要的算法。分治法通过将大问题分解为小问题,递归解决后合并结果;贪心算法在每一步选择局部最优解,追求全局最优;动态规划通过保存子问题的解,避免重复计算,确保全局最优。这三种算法各具特色,适用于不同类型的问题,合理选择能显著提升编程效率。
51 2
|
2月前
|
数据可视化 搜索推荐 Python
Leecode 刷题笔记之可视化六大排序算法:冒泡、快速、归并、插入、选择、桶排序
这篇文章是关于LeetCode刷题笔记,主要介绍了六大排序算法(冒泡、快速、归并、插入、选择、桶排序)的Python实现及其可视化过程。
22 0
|
4月前
|
搜索推荐 算法
十大排序算法-快排-希尔-堆排-归并-冒泡-桶排-选择-插入-计数-基数
十大排序算法-快排-希尔-堆排-归并-冒泡-桶排-选择-插入-计数-基数
|
4月前
|
算法 搜索推荐
算法设计 (分治法应用实验报告)基于分治法的合并排序、快速排序、最近对问题
这篇文章是关于分治法应用的实验报告,详细介绍了如何利用分治法实现合并排序和快速排序算法,并探讨了使用分治法解决二维平面上的最近对问题的方法,包括伪代码、源代码实现及时间效率分析,并附有运行结果和小结。
|
5月前
|
搜索推荐 C++ Python
Python排序算法大PK:归并VS快速,谁才是你的效率之选?
【7月更文挑战第13天】归并排序** 使用分治法,稳定且平均时间复杂度O(n log n),适合保持元素顺序和并行处理。
40 5
|
5月前
|
算法 Python
Python算法高手进阶指南:分治法、贪心算法、动态规划,掌握它们,算法难题迎刃而解!
【7月更文挑战第10天】探索Python算法的精华:分治法(如归并排序)、贪心策略(如找零钱问题)和动态规划(解复杂问题)。通过示例代码揭示它们如何优化问题解决,提升编程技能。掌握这些策略,攀登技术巅峰。
147 2
|
5月前
|
算法 程序员 Python
算法小白到大神的蜕变之路:Python分治法、贪心、动态规划,一步步带你走向算法巅峰!
【7月更文挑战第9天】探索算法之旅,以Python解锁编程高手之路。分治法如二分查找,将复杂问题拆解;贪心算法解决活动选择,每次选取局部最优;动态规划求斐波那契数列,避免重复计算,实现全局最优。每一步学习,都是编程能力的升华,助你应对复杂挑战,迈向算法大师!
52 1
|
5月前
|
存储 算法 Python
Python算法界的秘密武器:分治法巧解难题,贪心算法快速决策,动态规划优化未来!
【7月更文挑战第9天】Python中的分治、贪心和动态规划是三大关键算法。分治法将大问题分解为小问题求解,如归并排序;贪心算法每步选局部最优解,不保证全局最优,如找零钱;动态规划存储子问题解求全局最优,如斐波那契数列。选择合适算法能提升编程效率。
75 1
|
5月前
|
存储 算法 Python
震撼!Python算法设计与分析,分治法、贪心、动态规划...这些经典算法如何改变你的编程世界!
【7月更文挑战第9天】在Python的算法天地,分治、贪心、动态规划三巨头揭示了解题的智慧。分治如归并排序,将大问题拆解为小部分解决;贪心算法以局部最优求全局,如Prim的最小生成树;动态规划通过存储子问题解避免重复计算,如斐波那契数列。掌握这些,将重塑你的编程思维,点亮技术之路。
83 1