欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

算法 | 归并排序

程序员文章站 2024-02-04 19:32:58
归并排序 归并排序算法的核心就是 “归并”,将两个有序的数列合并,形成更大的有序数组。 归并排序的原理 上面说了,归并排序的核心就是“归并”。如果排序一个数组,那么将数组从中间分成前后两部分,对前后两部分分别进行排序,然后再将排序好的合并在一起,那么这样整个数组就会成为更大的有序数组。例如下面示图: ......

归并排序


归并排序算法的核心就是 “归并”,将两个有序的数列合并,形成更大的有序数组。

归并排序的原理


上面说了,归并排序的核心就是“归并”。如果排序一个数组,那么将数组从中间分成前后两部分,对前后两部分分别进行排序,然后再将排序好的合并在一起,那么这样整个数组就会成为更大的有序数组。例如下面示图:

算法 | 归并排序

归并排序使用的思想是分治思想,即是分而治之。将复杂问题分解成两个或者多个规模相同或类似的子问题,然后继续细化,当子问题足够简单,能够被求解,那么复杂的问题也就能够被求解出来。

这里跟递归的思想很像。递归也是将大问题细化为小问题,找出终止条件和递推公式。分治算法一般都是用递归实现的。分治的思想是一种解决问题的处理思想,递归是一种编程技巧,两者不会冲突。

上面说了,归并排序能够使用递归实现,那么现在先写出递推公式和终止条件。

# 递推公式
merge_sort(a...b) = merge(merge_sort(a...k), merge_sort(k+1, b))

# 终止条件
a >= b,不再继续分解

这里解释下上面的公式,终止条件不细讲,这里表示的,索引 a 大于等于 索引 b,即是已经到达数组末尾,不能够再细分。

讲下递推公式,merge_sort(a...b),表示给索引 a 到 k 之间的数组进行排序。这里将分体转为为两个子问题,merge_sort(a...k)merge_sort(k+1...b),其中 k 表示原数组中间的位置。而 merge() 函数表示当两个将两个排好序的子数组合并,这样就能够使原数组实现排序。

现在将这个递推公式转化为代码(这里使用伪代码):

# nums 表示数组,a 为起始点,b 为数组末尾索引
def merge_sort(nums, a, b):
    # 终止条件
    if a >= b:
        return
    
    # 取中间位置
    k = (a + b) // 2
    # 分治递归
    merge_sort(nums, a, k)
    merge_sort(nums, k+1, b)
    merge(nums[a...b], nums[a...k], nums[k+1, b])

最后 merge() 函数的作用,就是将后面两个子数组合并好后放到 nums[a...b] 中。现在主要的问题是如何实现分解后排序合并?

先用一个简单的示例,如下示图:

算法 | 归并排序

先申请临时数组 temp,定义指针 i,j 分别指向 两个子数组的第一个元素。比较 nums[i],nums[j],较小的元素放入临时数组中,同时该元素的指针向右移动。

循环上面的过程,直到其中一个子数组的所有数据都放入临时数组中,这个时候,将另外一个数组剩下的部分也放到临时数组中。这个临时数组就是最终合并的结果,将其复制回原数组即可。

现在用伪代码实现这个过程:

def merge(nums[a...b], nums[a...k], nums[k+1, b]):
    # length 表示数组大小
    temp = [0] * length

    # 初始化指针,i, j, ti
    i, j, ti, = a, k+1, 0
    # 取小值放入临时数组中,
    while (i <= k and j <= b):
        if (nums[i]<=nums[j]):
            temp[ti]=nums[i]
            i+=1
        else:
            temp[ti]=nums[j]
            j+=1
        ti+=1

    # 合并时会出现一个数组全部放入临时数组中,
    # 另外一个数组还有剩余,这里将剩余部分也放到临时数组中
    if i < nums[a...k].length:
        for x in range(i, nums[a...k].length):
            temp[ti] = nums[x]
            ti += 1
    else:
        for x in range(j, nums[k+1...b].length):
            temp[ti] = nums[x]
            ti += 1

    # 如果需要复制回原数组,也可以直接复制
    return temp

以上本篇幅就是关于归并排序的主要内容。


欢迎关注微信公众号《书所集录》