【数据结构】二叉树的顺序结构及实现，堆，向上调整算法，向下调整算法，数组建堆...

杨利霞

【数据结构】二叉树的顺序结构及实现，堆，向上调整算法，向下调整算法，数组建堆算法，堆排序
2 p) g  h. q9 j+ R/ K4 E$ M7 z
8 n2 w" m. [; ]文章目录
1.二叉树的顺序结构
2.堆的概念及结构
3.堆的实现
3.1堆的总实现
3.2堆的向上调整算法---O(logN)
4 e5 z% q  h  w: e& I+ S+ a3.3堆的向下调整算法---O(logN)
4.数组建堆算法（建大堆）
; ~9 Z) v' Q: B4 v. |4.1向上调整建堆
$ ~$ d; D* |( v4.2向下调整建堆
% @  u% K$ e& b5.堆排序
4 \! t! |" r  l5 ~& [; ?1.二叉树的顺序结构
普通的二叉树不适合用数组来存储，因为它可能会存在大量的空间浪费，而完全二叉树更适合使用顺序结构存储。现实中我们通常通过堆（堆是一种二叉树的结构），使用顺序结构的数组来存储，需要注意的是这里的堆和操作系统虚拟进程地址空间中的堆是两回事，一个是数据结构，一个是操作系统中管理内存的一块区域分段。

' t- K. J9 q0 _# a
/ V  i* T9 [$ t
2.堆的概念及结构
如果有一个关键码的集合K={k0,k1,k2,…,Kn-1},把它的所有元素按完全二叉树的顺序表存储方式存储在一个一维数组中，并满足以下情况：
% D; j: m* h+ l0 r+ Q$ O" A
  ~% P, k8 Y% o$ Q1.堆中某个节点的值总是不大于或不小于其父节点的值；
% i. J/ V% q- u/ ~0 m) i6 d+ ~' y! I2.堆总是一棵完全二叉树。

' ?$ J+ S. j5 B5 h
9 y0 H: d% B; X. J( i3.堆的实现
3.1堆的总实现
' O1 }. |  l2 L2 n( {堆的实现代码（全），请点击——>堆的实现代码

6 K6 f, P7 N9 s3.2堆的向上调整算法—O(logN)
% e* l; E8 }5 R& O6 w* y# M7 c使用场景：向堆中插入数据，需要使用向上调整算法，因为向堆中插入数据是将数据插入到下标为size的位置，插入一个数据，size++，此时可能就不满足小堆（或大堆），因此要对其进行调整，此处以小堆为例，向上调整算法只需要从插入的结点位置开始和父节点比较（一路和祖先进行比较），若a[child]<a[parent],则交换，若a[child]>=a[parent],则满足小堆，直接break，跳出循环。同时，循环结束的条件是child==0,因为此时小数已经到达堆顶，调整完成。—一遍建堆，一遍调整
3 ~: [* n) ^* o6 B
+ b( q1 s3 M( p2 Y' {) f' V  G0 b代码实现：（以建小堆为例）
! x0 A% y0 u1 c. [1 w
/ l9 r  H8 h1 G. ^. S* ]: x: Bvoid Swap(HPDataType* e1, HPDataType* e2)
6 e; f/ H5 }8 l% y) x; i{
* Y% y; q7 }% D. L9 ]8 R0 R        HPDataType tmp = *e1;
! I/ R4 X3 B/ m3 S; w        *e1 = *e2;
        *e2 = tmp;
5 G- S6 e& V4 T% b0 p1 E}
void AdjustUp(HPDataType* a,int child)
9 _9 Z; z0 ~$ [8 p{
        int parent = (child - 1) / 2;
# Q, D* s2 Z- W; L( N        while (child > 0)
        {
" j' a8 [% h" I8 R                if (a[child] < a[parent])
1 R, W$ C' k, y. O2 s. F: F                {
  H1 Y4 k5 t1 @# n% B8 x$ k                        Swap(&a[child], &a[parent]);
1 Z% y) j& y+ a! m0 \( K! R; j                        //向上调整
/ N  P$ e$ |. h, |+ X" |% ~7 \                        child = parent;
- {8 o4 O) J. p4 ~9 @7 `9 d; u                        parent = (child - 1) / 2;
+ X+ d: z# e5 l- q8 b3 V& V                }
5 q* |0 p4 x9 S8 x5 j                else//满足小堆条件
6 f/ Z7 P4 _% k3 r$ X! s                {
                        break;
                }
        }
: d. c; b: y- F$ E" g4 U) `/ H, l}
! e0 }; I: [" W. @$ \9 m3 `) W' u//插入x，继续保持堆形态
6 Q5 Q4 g; b+ k) t" Vvoid HeapPush(HP* php, HPDataType x)//插入数据到堆中
{
        assert(php);
% a  o+ p' L$ W8 L        //扩容
* [' O5 M5 _' A0 U  ]        if (php->size == php->capacity)
        {
: ]( G- C/ a) ]* e; k  c' a3 O                int newcapacity = php->capacity == 0 ? 4 : php->capacity * 2;
) U; z: g* L8 a# M8 [                HPDataType* tmp = (HPDataType*)realloc(php->a, newcapacity * sizeof(HPDataType));
4 {2 z  m7 Z$ b: I1 _) a$ S" R                if (tmp == NULL)
9 \0 r/ n3 X: \; L; ~                {
$ J* T$ [; v! _                        perror("realloc fail");
                        exit(-1);
                }
                php->a = tmp;
$ w' c* [. _. ~" d6 w8 V# J                php->capacity = newcapacity;
5 g* Y$ f4 k" H$ \8 [8 A+ y        }
" F. O# w+ D4 H8 D        php->a[php->size] = x;
: K: ]1 P8 K0 j5 ^; j: b        php->size++;

6 n$ o4 r: n$ b' M        //向上调整函数
        AdjustUp(php->a, php->size - 1);
$ }/ P% h) m5 Y/ @  {; E7 e}
$ \( o/ x3 C& f3 k+ o
4 a# S0 J/ X* d1
2
3
5 L9 A$ S3 z. N% n4
2 u3 U. |: J% u5 y; f3 w+ j% s5
6
7
' ~+ m, ?) q) w8
- L6 Z, c. q5 L. s  a! ^9
10
11
9 a( a# L( @$ A% F1 S; j12
13
- t6 c: g* m2 c4 D4 ^14
  {) n$ n' S, M3 d) t+ {. s15
16
' Y' I& I) ^0 M' S/ E: R17
18
/ `5 x/ B/ L9 u& m19
% W% \0 ~0 p# k1 D9 z! U20
6 u2 P* y" H8 J21
. R. z$ G+ p0 `% s7 n$ _, E22
( U8 z2 A( c) t23
# t5 ?: p  Y* q0 t1 M0 n, ?24
25
, s6 h3 e* p7 \0 s26
27
% P; v+ X0 e  R- }9 H  I5 Y28
  g' R7 K; ~( j: O' w29
30
% e5 Y) C7 f4 A/ ?: L, ~9 r' t31
+ i3 e' x( _% @7 g4 q, P  d32
33
34
* V$ P+ I2 s- G% t* |  l6 {) |35
36
37
' X% r% _6 B5 ~38
) s+ n/ E2 l+ h: E* g39
' B; `2 o1 {/ l" Q40
3 ~) n+ Y$ ^4 n4 L0 v" S41
42
43
; }5 m& s- g8 w% }. a' `44
45
6 U) f4 ^5 w6 p) b. v  I46
47
8 |( }0 O, f% ?, }0 L& \) g1 k

3.3堆的向下调整算法—O(logN)
# g0 J, c* ^. P6 k. \# {" k1.使用场景：删除堆顶元素HeapPop，需要用到堆的向下调整算法。
- _6 a5 C/ ?- @删除堆顶元素，作用：找次大的数或者次小的数—时间复杂度O(logN)
方法：在删除堆顶元素时用到了向下调整算法，建立了一个小堆，如果我们使用向前挪动数组元素的方法，删除堆顶元素，时间复杂度为O(N);如果先将堆顶元素和最后一个元素交换，然后size–，删除这个元素，再用向下调整算法，时间复杂度就是O(logN)。
) R* ^. o% G, z7 u2.向下调整算法的前提是：当前的树左右子树必须都是一个堆
4 o$ E% w% S' R/ x* k/ V% F; V8 f  K3.算法的核心思想：从根结点开始，选出左右孩子中小的那一个，跟父亲交换，因为是建小堆（原来的堆是小堆），所以小的往上交换，大的往下沉，如果要建大堆则相反。
+ T2 J+ o# \4 X* b0 V" E3 ~
下图是在利用向下调整算法建一个小堆：


代码实现：（建小堆为例）

//向下调整算法---O(logN)
9 q: L* p: I) o* q( j: j7 gvoid AdjustDown(HPDataType* a,int size,int parent)
{
        //先假设左边的孩子是最小的
" d# y3 z% P9 L: M4 \6 _        int minchild = parent * 2 + 1;
        while (minchild < size)
        {
: J9 N  o1 {4 v4 l6 T# o* A                //找出小的那个孩子进行交换
                if (minchild+1<size && a[minchild + 1] < a[minchild])
  l& x# s$ G$ n                {
0 W$ q/ S$ D6 n                        minchild += 1;
9 \$ Z7 s5 [/ ^5 X4 \1 Y5 O                }
: ?" n/ l& U" `1 h                if (a[minchild] < a[parent])
& N" m4 \9 R5 I7 e0 f6 s                {
, ]) d2 d. s; u# j4 ?1 W                        Swap(&a[parent], &a[minchild]);
$ R- {" D, P, m0 r5 f' W$ }& h                        parent = minchild;
                        minchild = 2 * parent + 1;
                }
5 `9 }+ Q. ]5 v                else
                {
                        break;
  R; Z& |- V8 P! S9 ^1 U* o8 m1 g                }
0 J# u4 S+ n# }$ p        }
3 v1 T2 P9 }" P0 A}
1 B2 W+ Y: N& A9 p' J  ~void HeapPop(HP* php)//删除堆中的数据。保持堆形态
{
+ d) O$ o' ]5 o1 `, e9 q! a        assert(php);
0 e3 e; }" |4 L7 z7 \6 q        assert(!HeapEmpty(php));
9 L% A5 v* [+ @+ f$ G        //方法：交换堆顶和最后一个元素
( b, Z  Q+ R7 x1 ^2 U9 }) G        //如果用挪动的方式，时间复杂度为O(N),而用向下调整算法，时间复杂度只有O(logN)
        Swap(&php->a[0], &php->a[php->size - 1]);
        php->size--;

6 t9 t. Q" @0 p, }        AdjustDown(php->a, php->size, 0);//从堆顶的位置开始，向下调整
) m) ]; J- }6 d; G0 c+ r0 r}
4 v9 O  w. _+ y  e) U2 _
1
2
3
4
1 W- M# e7 Z2 n# E& g7 ~5
* Q2 W2 ~. \9 u7 c6
3 @/ E+ n8 s$ r8 o1 N7
8
. R+ w) v+ L& K8 J" c- N* B* j9
$ R1 N! [5 x7 Y2 T% X  M10
11
12
13
* l! L3 u; [; e, u$ S14
15
16
17
9 E' q2 K1 d# S; u2 x% M8 k/ C: Y18
19
. @& w6 X0 i( E4 m7 d& r20
* u5 T( ?& U" |. T. u& Y21
* ~, i$ `$ S! v- e% [& Q22
0 T$ {9 S" b& i1 V+ q4 R6 T* H23
) H& n) e+ r' J8 n5 w/ H* z, P24
25
, f2 ~: E- B+ ]9 n9 s26
% p1 a. \# \' _1 E$ ^27
% ^, y+ \) p7 _3 o6 Y28
, u: ]  K& v/ C2 ]1 X29
5 o9 k) v5 x2 [* A" R2 t2 _30
31
32
0 H( S9 }: J* L- Z/ D33
; Y/ Y2 c- t& L% b! d34
0 e* r/ b2 S. s/ u, A6 v8 C35
4.数组建堆算法（建大堆）
下面我们给出一个数组，这个数组逻辑上可以看做一棵完全二叉树，但是还不是一个堆，现在我们通过算法，把它构建成一个堆。那么根结点的左右子树不是堆，我们怎么调整呢？这里介绍两种算法，优先使用向下调整算法，时间复杂度更优。

7 n) B& G, L4 J数组建堆代码的实现，请点击——>数组建堆算法
; s! ~- ?5 q# e2 k) ?9 p9 _8 w1 K
4 y6 ?1 t5 L6 g# U3 T4.1向上调整建堆
在原数组的基础上（物理上），第一个元素保持不动，之后的元素类似HeapPush,堆的插入，向上调整建堆。向上调整建堆的时间复杂度分析：O(N*logN)


/ z, N' {9 f2 u7 G3 L' ]代码实现：

#include<stdio.h>
void Swap(int* e1, int* e2)
{
        int tmp = *e1;
        *e1 = *e2;
. ]( S$ B5 _9 s  C& ]+ K; p+ J        *e2 = tmp;
0 \! j" q% V1 E$ v8 Q* B}
void AdjustUp(int* a, int child)
{
) W* f+ U! Y/ J  y! |        int parent = (child - 1) / 2;
4 @! P8 h, `! i$ M% ~$ `. m        while (child>0)
+ R# T  N4 l( U8 v' m$ j        {
                if (a[child] > a[parent])//建大堆
2 T# _& Z; v( r3 r( M) a5 |                {
3 }. z2 c2 K$ b" e                        Swap(&a[child], &a[parent]);
5 G" f; Z% q- O5 ]                        child = parent;
                        parent = (child - 1) / 2;
                }
                else
                {
) _# N" _' F% [/ a& ~                        break;
                }
# _8 M9 K" q$ L( G1 k5 T        }
& q2 P, h1 y3 U3 F* i! g) w}
//向上调整算法建堆
void HeapCreate(int* a, int n)
3 l& N1 w( g0 E. z# Q  q. ?- E{
' N! Z: J+ u$ P( ]        //向上调整建堆,第一个元素保持不动
        for (int i = 1; i < n; i++)
) i. e( q! {0 {, B; c; X- i4 ]        {
+ i$ d! j: X+ y% O                AdjustUp(a, i);//物理上是个数组，在数组的基础上建堆,从第2个数据开始插入，一边插入一边建堆
        }
}
: C; J( C) C% w5 R& L, ?5 \void HeapPrint(int* a, int n)
{
        for (int i = 0; i < n; i++)
) q$ q+ C; G- d; m$ R* C        {
2 T. r  ?' z3 s7 `* B                printf("%d ", a);
        }
7 w+ \' v5 B9 V& C9 `}
4 r7 h6 a0 ]( n6 r3 gint main()
+ G* x8 K3 I! m2 B9 r2 v: v! `{
' j  ?' }! B! a+ v. v3 J        int a[] = { 65,100,60,32,50,70 };
        //int a[] = { 15,1,19,25,8,34,65,4,27,7 };

0 z2 d* r: r  b7 z% Z" J        HeapCreate(a, sizeof(a) / sizeof(a[0]));
- Z, \# k, F$ R6 q( V9 \        HeapPrint(a, sizeof(a) / sizeof(a[0]));
2 \1 k$ Z& o  v/ q% n}

5 E8 R8 i9 z6 @- k4 @* P: S1
. u( x4 ]# B7 C  N; }2
; T, X! {* w9 x( z& d/ @0 X- }3
9 k7 v1 J; e2 U: ]2 L. [4
" D$ i; P! v5 N5
4 h$ G5 \/ p; J4 M  g6
7
8
9
10
# V3 H- Z% b! |8 g' Q# c; A; K11
. L) l5 j  _9 m! B& b7 R12
13
( i9 w* r" u* d  Z4 A2 x3 A% w) f14
2 f4 H  Y  [" a: k15
16
3 \# B4 z4 p1 ~: K0 b9 Q2 Q17
5 y$ y# v6 w7 v0 ^5 S( W9 j) I18
19
20
0 I, ?7 m8 U6 ]21
$ H9 h0 `+ `- p! T22
23
! V$ b- j% b+ F1 Y24
5 r5 l4 ]+ X; i3 v- _& z25
5 G& r: e/ s# B) {26
8 q) y+ Z" p, D27
28
0 H/ @! e  L7 p+ i( }29
30
% g8 X6 E2 N1 r# K* N31
5 p& }& E2 W+ o4 ^7 ]! m, b" A32
: _! j3 P/ g1 P- s33
34
35
36
) p: |; R+ l0 C5 [3 N: _37
+ U3 M) `- i( l38
; A5 S4 C4 r, H+ ?& ?, O! [/ ?39
  T0 U" l% d) a# I40
41
. q& G! |% s5 b7 D( H) x/ n* J; y' t. D42
43
44
4 G. d3 D$ K# [0 f, f# B45
46
47
48
4.2向下调整建堆
向下调整建堆，从倒数第一个非叶子结点（最后一个结点的父亲）开始向下调整，调整完后，下标–，到另一个子树调整，直到调整到根。时间复杂度分析：O(N),优于向上调整建堆，分析见下图：


3 b. l2 ~" D% c8 f
3 X2 L0 ]% E* k& l6 K代码实现：

void AdjustDown(int* a, int size, int parent)
{
        //先假设左边的孩子是最小的
% e1 Z0 L; U; a$ Y        int minchild = parent * 2 + 1;
        while (minchild < size)
        {
1 q' E6 x: K6 N  P+ @+ _+ |5 s$ D                //找出小的那个孩子
* ~/ h( t& E7 f% q; u. m                if (minchild + 1 < size && a[minchild + 1] > a[minchild])
                {
/ M, {' E7 [1 F  d                        minchild ++;
                }
                if (a[minchild] > a[parent])
                {
3 q1 U6 s$ e" A1 y. J. V                        Swap(&a[minchild], &a[parent]);
# p/ K1 @0 f/ O0 R) T                        parent = minchild;
9 [( X+ \2 A: k: Z. o& f! c                        minchild = 2 * parent + 1;
                }
8 S; ^: Y3 q$ ^' A9 _* ~) B                else
' z* h& E( P5 x                {
; M" \9 G: }: w6 E+ J0 }# S                        break;
                }
        }
}
void HeapCreate(int* a, int n)
4 l5 p% i; t6 T7 E1 q: _{
        //向上调整建堆,第一个元素保持不动
        //for (int i = 1; i < n; i++)
& F9 b6 A5 |, f        //{
+ V, ?6 Z) m0 X        //        AdjustUp(a, i);//物理上是个数组，在数组的基础上建堆,从第2个数据开始插入，一边插入一边建堆
        //}
" Z9 j4 f% R/ Q8 ^( K        //向下调整建堆
% \7 N5 `& k6 i, N, i        //n-1是最后一个数据的下标，再-1/2算的就是最后一个结点的父节点
0 c% G0 |/ p# e        for (int i=(n - 1 - 1)/2;i>=0; i--)
        {
) @4 Q5 w7 J8 c" [; i: d2 u0 O4 H) e                AdjustDown(a, n, i);
6 b" k% B8 P1 o+ K% v( }        }
}

1
2
2 m7 H7 ^" c, i+ U3
4
5
6
7 X; X# G1 B* k, M4 @2 x" F4 G7
8
9
, F# i! M( @; C* Y) {) d10
. j1 [- u* P0 i4 X" v11
0 f+ Z0 H2 X1 o- r) ^; D6 X9 s7 b12
+ D3 g/ C0 R4 Z; w2 y13
/ _, H3 e2 x: T. ]/ v) e- a14
. Y8 W- v; B0 o" C& o15
3 X. [  d4 Y# ^0 L16
! v' l0 x. R$ L" J: M17
18
19
, d3 x3 V+ n9 D0 h4 }20
3 B9 W3 E9 I- D: u* d' W21
22
23
& U* N: Y; [7 u- y24
25
26
; C& \- O( d/ r) h. _27
28
29
30
31
32
# u# ~* C1 I7 \. ~* Z33
: k5 v; h  c" e* c0 \! n34
1 x2 s/ S9 y6 L$ R" `  }0 Z+ X35
  k/ }$ q! E3 N" ]5 V' s& Z36
7 F! A. J3 K- O$ V/ w/ I37
5.堆排序
堆排序实现代码，请点击——>堆排序代码
! q% H+ `$ B5 r+ F* M8 v
1.堆排序的大思路，也是一种选择排序。大多数学生认为如果是排升序，建的是小堆，排降序建的是大堆，其实结果是相反的。
升序—建大堆
降序—建小堆
9 e3 }, C$ A/ R5 U) ?' F2.推论：假设我们排升序，建的是小堆，思路是，先用向下调整算法建一个小堆，时间复杂度为O(N),堆顶就是第一次找出的最小数，然后再将剩下的数据看成堆，重新建堆，选次小的数据。这种算法可以是可以，但是太多此一举了，并且整体下来的时间复杂度为O(N^2)。
假设升序—建大堆，思路是，先把一开始建好的大堆，第一个数据和最后位置交换，这样就排好了最大的那个数据，然后不把最后一个数据看作堆里面的，继续向下调整建堆，选出次大的，与倒数第二个数据交换，后续一次处理。这样整体下来的时间复杂度就是O(N*logN)
* h# |% ~: h8 w$ m+ n9 c
void HeapSort(int* a, int n)
" d1 m* c9 |3 X9 V+ Y{
/ m8 A8 j, Z2 K        //堆排序的大思路，也是一种选择排序
        //升序---大堆
        //降序---小堆
, X2 M# X- ?% g& E$ O9 a        //先建堆：向下调整建堆---O(N)
! L, ?& t: G3 i        for (int i = (n - 1 - 1) / 2; i >= 0; i--)
$ R$ c+ ]# p' C" c: Z        {
                AdjustDown(a, n, i);
; g8 T5 F# M  c7 I! n        }

& A( [& b+ v  t* @        //此时堆顶是最大元素，交换堆顶元素与最后一个元素
        //再将剩下的n-1个元素重新向下调整，找到次大的
" J! {9 \2 g- {3 J        int i = 1;
, _, Z# |" m% ^! C        while (i < n)
        {
                Swap(&a[0], &a[n - i]);
                AdjustDown(a, n-i, 0);
                i++;
        }
}

————————————————
( T3 D7 Z2 @7 _+ M% O7 P版权声明：本文为CSDN博主「SouLinya」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_63449996/article/details/126775501