【算法基础】 - 归并排序

本文旨在对于个人知识的梳理以及知识的分享，如果有不足的地方，欢迎大家在评论区指出

核心思想

归并排序的核心思想主要有以下几步：

• 找到区间中点mid
• 将[l, mid]以及[r, mid]分别进行排序
• 将这两个区间合并为一个区间

时间复杂度分析

归并排序的时间复杂度为$O(nlogn)$

模板代码

Go

package main

import "fmt"

func merge_sort(a []int, l int, r int) {
	if l >= r {
		return
	}
	// 找中点
	mid := (l + r) >> 1
	merge_sort(a, l, mid)
	merge_sort(a, mid+1, r)

	hp := make([]int, r-l+1)
	i, j, idx := l, mid+1, 0
	for i <= mid && j <= r {
		if a[i] < a[j] {
			hp[idx] = a[i]
			idx += 1
			i += 1
		} else {
			hp[idx] = a[j]
			idx += 1
			j += 1
		}
	}
	for i <= mid {
		hp[idx] = a[i]
		idx += 1
		i += 1
	}
	for j <= r {
		hp[idx] = a[j]
		idx += 1
		j += 1
	}
	for k:=0; k<idx; k++ {
		a[k+l] = hp[k]
	}
}

func main() {
	var n int
	fmt.Scanf("%d", &n)
	a := make([]int, n)
	for i:=0; i<n; i++ {
		fmt.Scanf("%d", &a[i])
	}

	merge_sort(a, 0, n-1)
	for i:=0; i<n; i++ {
		fmt.Printf("%d ", a[i])
	}
}

例题扩展

• 逆序对的数量