|
| 1 | +package sort |
| 2 | + |
| 3 | +func HeapSort(a []int) []int { |
| 4 | + // 1、无序数组a |
| 5 | + // 2、将无序数组a构建为一个大根堆 |
| 6 | + for i := len(a)/2 - 1; i >= 0; i-- { |
| 7 | + sink(a, i, len(a)) |
| 8 | + } |
| 9 | + // 3、交换a[0]和a[len(a)-1] |
| 10 | + // 4、然后把前面这段数组继续下沉保持堆结构,如此循环即可 |
| 11 | + for i := len(a) - 1; i >= 1; i-- { |
| 12 | + // 从后往前填充值 |
| 13 | + swap(a, 0, i) |
| 14 | + // 前面的长度也减一 |
| 15 | + sink(a, 0, i) |
| 16 | + } |
| 17 | + return a |
| 18 | +} |
| 19 | +func sink(a []int, i int, length int) { |
| 20 | + for { |
| 21 | + // 左节点索引(从0开始,所以左节点为i*2+1) |
| 22 | + l := i*2 + 1 |
| 23 | + // 有节点索引 |
| 24 | + r := i*2 + 2 |
| 25 | + // idx保存根、左、右三者之间较大值的索引 |
| 26 | + idx := i |
| 27 | + // 存在左节点,左节点值较大,则取左节点 |
| 28 | + if l < length && a[l] > a[idx] { |
| 29 | + idx = l |
| 30 | + } |
| 31 | + // 存在有节点,且值较大,取右节点 |
| 32 | + if r < length && a[r] > a[idx] { |
| 33 | + idx = r |
| 34 | + } |
| 35 | + // 如果根节点较大,则不用下沉 |
| 36 | + if idx == i { |
| 37 | + break |
| 38 | + } |
| 39 | + // 如果根节点较小,则交换值,并继续下沉 |
| 40 | + swap(a, i, idx) |
| 41 | + // 继续下沉idx节点 |
| 42 | + i = idx |
| 43 | + } |
| 44 | +} |
| 45 | +func swap(a []int, i, j int) { |
| 46 | + a[i], a[j] = a[j], a[i] |
| 47 | +} |
0 commit comments