蓝桥杯Python——day2归并排序

归并排序：分治

待排序数组为nums, 左边界为l，右边界为r

1. 确定分界点：mid = （l + r）>> 1
2. 递归分隔
3. 合并：双指针法

def merge_sort(nums, l, r) :
    if l >= r :
        return
    mid = (l + r) >> 1
    merge_sort(nums, l, mid)
    merge_sort(nums, mid + 1, r)
    tmp = []
    i, j = l, mid + 1
    while i <= mid and j <= r :
        if nums[i] <= nums[j] :
            tmp.append(nums[i])
            i += 1
        else :
            tmp.append(nums[j])
            j += 1

    if i <= mid :
        tmp += nums[i: mid + 1]
    elif j <= r :
        tmp += nums[j : r + 1]
    nums[l : r + 1] = tmp

tips : py列表切片赋值一列表，列表元素转换为原列表中元素
字符串的join方法，可以用来用字符串来连接列表中的字符串元素

例题

给定一个长度为 n 的整数数列，请你计算数列中的逆序对的数量。

逆序对的定义如下：对于数列的第 i 个和第 j 个元素，如果满足 i<j 且 a[i]>a[j]，则其为一个逆序对；否则不是。

输入格式
第一行包含整数 n，表示数列的长度。

第二行包含 n 个整数，表示整个数列。

输出格式
输出一个整数，表示逆序对的个数。

数据范围
1≤n≤100000，
数列中的元素的取值范围 [1,109]。

输入样例：
6
2 3 4 5 6 1
输出样例：
5

n = int(input())

nums = list(map(int, input().strip().split()))

res = 0
def merge_sort(nums, l, r) :
    global res
    if l >= r :
        return
    mid = (l + r) >> 1
    i, j = l, mid + 1
    tmp = []
    merge_sort(nums, l, mid)
    merge_sort(nums, mid + 1, r)
    while i <= mid and j <= r :
        if nums[i] <= nums[j] :
            tmp.append(nums[i])
            i += 1
        else:
            tmp.append(nums[j])
            j += 1
            res += mid - i + 1
    if i <= mid :
        tmp += nums[i : mid + 1]
    elif j <= r :
        tmp += nums[j : r + 1]
    nums[l : r + 1] = tmp


merge_sort(nums, 0, n - 1)

print(res)

tips : python函数中要引用外部的全局变量，需要在函数中使用global关键字声明

总结：快排属于尾递归，对应于树的遍历属于前序遍历，归并排序是首递归，对应于后序遍历