• 树、二叉树、堆及其应用（堆排序、top-k问题）

    目录

    树的概念与结构

    概念：

    与树相关的概念:

    树的表示：

    二叉树

    概念：

    特殊的二叉树：

    二叉树性质：

    二叉树的存储结构：

    堆

    堆的概念:

    堆的实现：

    堆的创建：

    堆的插入：

    堆的删除：

    堆的应用

    堆排序：

    TOP-K问题：

    树的概念与结构

    概念：

    树是一种非线性的数据结构，它是由n（n>=0）个有限结点组成一个具有层次关系的集合。把它叫做树是因为它看起来像一棵倒挂的树，也就是说它是根朝上，而枝叶朝下的。

    • 每一棵树都有一个特殊的结点，称为根结点，根节点没有前驱结点
    • 除根节点外，其余结点被分成M(M>0)个互不相交的集合T1、T2、……、Tm，其中每一个集合Ti(1<= i<= m)又是一棵结构与树类似的子树。每棵子树的根结点有且只有一个前驱，可以有0个或多个后继。
    • 树是递归定义的。

     需要注意的是：树形结构中，子树之间不能有交集。

    如上图一中B节点和A节点有交集，图二中A节点和C节点有交集，图三中C节点和D节点有交集，因此三个结构都不是树形结构。

    与树相关的概念:

    • 节点的度：一个节点含有的子树的个数称为该节点的度； 如上图：A的度为6
    • 叶节点或终端节点：度为0的节点称为叶节点； 如上图：B、C、H、I...等节点为叶节点
    • 非终端节点或分支节点：度不为0的节点； 如上图：D、E、F、G...等节点为分支节点
    • 双亲节点或父节点：若一个节点含有子节点，则这个节点称为其子节点的父节点； 如上图：A是B的父节点
    • 孩子节点或子节点：一个节点含有的子树的根节点称为该节点的子节点； 如上图：B是A的子节点
    • 兄弟节点：具有相同父节点的节点互称为兄弟节点； 如上图：B、C是兄弟节点
    • 树的度：一棵树中，最大的节点的度称为树的度； 如上图：树的度为6
    • 节点的层次：从根开始定义起，根为第1层（也可以是第0层），根的子节点为第2层，以此类推；
    • 树的高度或深度：树中节点的最大层次； 如上图：树的高度为4
    • 堂兄弟节点：双亲在同一层的节点互为堂兄弟；如上图：H、I互为兄弟节点
    • 节点的祖先：从根到该节点所经分支上的所有节点；如上图：A是所有节点的祖先
    • 子孙：以某节点为根的子树中任一节点都称为该节点的子孙。如上图：所有节点都是A的子孙
    • 森林：由m（m>0）棵互不相交的树的集合称为森林；

    树的表示：

    树结构相对线性表比较复杂，要存储表示起来就比较麻烦，既然保存值域，也要保存结点和结点之间的关系，实际中树有很多种表示方式如：双亲表示法，孩子表示法、孩子双亲表示法以及孩子兄弟表示法等。其中最常用的是孩子兄弟表示法。                                                                                结构代码如下：

    1. typedef int DataType;
    2. struct Node
    3. {
    4. 	struct Node* child; //孩子结点
    5. 	struct Node* brother; //指向其下一个兄弟结点
    6. 	DataType data; //数据域
    7. };

    二叉树

    概念：

    顾名思义，就是每个节点最多只有两个分叉即每个父节点至多有两个子节点的树形结构。

    • 二叉树不存在度大于2的结点
    • 二叉树的子树有左右之分，次序不能颠倒，因此二叉树是有序树

    注意：对于任意二叉树，都可以由以下情况复合而成：

    特殊的二叉树：

    满二叉树：一个二叉树，如果每一个层的结点数都达到最大值，则这个二叉树就是满二叉树。也就是说，如果一个二叉树的层数为K（根节点层数为1），且结点总数是 2^k-1，则它就是满二叉树。

    完全二叉树：完全二叉树是由满二叉树而引出来的，满二叉树是一种特殊的完全二叉树。对于一个层数为k（根节点层数为1）的二叉树，前k-1层为满二叉树，第k层中所有节点从左至右是连续的，则此二叉树为完全二叉树。高度为h的完全二叉树节点个数范围为：[2^(h-1)+1 , 2^h-1]。

    二叉树性质：

    1. 若规定根节点的层数为1，则一棵非空二叉树的第i层上最多有2^(i-1) 个结点.
    2. 若规定根节点的层数为1，则深度为h的二叉树的最大结点数是 2^h-1.
    3. 对任何一棵二叉树, 如果度为0的叶结点个数为 n0, 度为2的分支结点个数为 n2,则有 n0＝n2＋1
    4. 若规定根节点的层数为1，具有n个结点的满二叉树的深度，h=log2(n+1) . (log2(n+1) 是以2
    为底，(n+1)的对数)
    5. 对于具有n个结点的完全二叉树，如果按照从上至下从左至右的数组顺序对所有节点从0开始编号，则对于序号为i的结点有：
            1. 若i>0，i位置节点的双亲序号：(i-1)/2；i=0，i为根节点编号，无双亲节点
            2. 若2i+1=n否则无左孩子
            3. 若2i+2=n否则无右孩子

    二叉树的存储结构：

    二叉树一般可以使用两种结构存储，一种顺序结构，一种链式结构。

    • 顺序存储

    顺序结构存储就是使用数组来存储，一般使用数组只适合表示完全二叉树，因为不是完全二叉树会有空间的浪费。而现实中只有堆才会使用数组来存储。二叉树顺序存储在物理上是一个数组，在逻辑上是一颗二叉树。

    观察数据存储和节点间的关系，可以发现：

    父节点下标=(相应子节点下标-1)/2（子节点不分左右）；

    左孩子下标=相应父节点下标*2+1；

    右孩子下标=相应父节点下标*2+2。

    • 链式存储

    即用链表来表示一棵二叉树，用指针来指示元素的逻辑关系。 通常的方法是链表中每个结点由三个域组成，数据域和左右指针域，左右指针分别用来指出该结点左孩子和右孩子所在的结点的存储地址 。链式结构又分为二叉链和三叉链。

    1. typedef int BTDataType;
    2. // 二叉链
    3. struct BinaryTreeNode_2
    4. {
    5. 	struct BinTreeNode* Left; // 指向当前节点左孩子
    6. 	struct BinTreeNode* Right; // 指向当前节点右孩子
    7. 	BTDataType data; // 当前节点值域
    8. }
    9. // 三叉链
    10. struct BinaryTreeNode_3
    11. {
    12. 	struct BinTreeNode* Parent; // 指向当前节点的父节点
    13. 	struct BinTreeNode* Left; // 指向当前节点左孩子
    14. 	struct BinTreeNode* Right; // 指向当前节点右孩子
    15. 	BTDataType data; // 当前节点值域
    16. }；

    堆

    相对于普通二叉树，完全二叉树更适合使用顺序结构存储。而现实中通常用顺序结构储存堆(和操作系统中存储数据的堆区不是一回事，此处的堆是一种存储数据的结构，也是一种二叉树)。

    堆的概念:

    一个数集中的所有元素按照完全二叉树的顺序存储方式存储在一个一维数组中，并满足：在二叉树结构中任一父节点值都大于等于子节点（任一父节点值都小于等于子节点值），则称为大堆或大根堆（小堆或小根堆）。

    堆的性质：

    • 堆中某个节点的值总是不大于或不小于其父节点的值；
    • 堆是一棵完全二叉树。

    堆的实现：

    堆的创建：

    想要创建一个堆，有两种方法：

    向上调整建堆 (前边的数据必须是堆)  

    向下调整建堆 (左右子树必须是堆)

    给定一个数组arr，arr数组逻辑上是一个完全二叉树，但还不是一个堆，存储顺序完全是随机的，当我们想要把arr数组变为一个堆时，也是可以用两种方法：

    1、向下调整（时间复杂度：O(N)、空间复杂度：O(1)（不计算堆所占的空间））：因为向下调整时，要保证左右子树是堆，因此我们可以从数组倒数第一个非叶子节点开始，逐个元素向下调整，一直调整到根节点。

    代码如下：

    1. typedef int HPDataType;
    2. typedef struct Heap
    3. {
    4. 	HPDataType* a;
    5. 	int size;
    6. 	int capacity;
    7. }HP;
    8.  
    9. //打印
    10. void HeapPrint(HP* php)
    11. {
    12. 	assert(php);
    13. 	int i = 0;
    14. 	for (i = 0; i < php->size; i++)
    15. 	{
    16. 		printf("%d ", php->a[i]);
    17. 	}
    18. 	printf("\n");
    19. }
    20.  
    21. //交换函数
    22. void Swap(HPDataType* p1, HPDataType* p2)
    23. {
    24. 	HPDataType k = *p1;
    25. 	*p1 = *p2;
    26. 	*p2 = k;
    27. }
    28.  
    29. //向下调整(小堆)
    30. void AdjustDown(HPDataType* a, int n, int parent)
    31. {
    32. 	assert(a);
    33. 	int child = parent * 2 + 1;
    34. 	while (child < n)
    35. 	{
    36. 		//找出小的
    37. 		if (child + 1 < n && a[child] > a[child + 1])
    38. 		{
    39. 			child += 1;
    40. 		}
    41.  
    42. 		if (a[child] < a[parent])
    43. 		{
    44. 			Swap(&a[child], &a[parent]);
    45. 			parent = child;
    46. 			child = parent * 2 + 1;
    47. 		}
    48. 		else
    49. 		{
    50. 			break;
    51. 		}
    52. 	}
    53. }
    54.  
    55. //初始化
    56. void HeapInitArray(HP* php, HPDataType* a, int n)
    57. {
    58. 	assert(php);
    59. 	assert(a);
    60. 	php->a = (HPDataType*)malloc(sizeof(HPDataType) * n);
    61. 	if (php->a == NULL)
    62. 	{
    63. 		perror("malloc failed");
    64. 		exit(-1);
    65. 	}
    66. 	php->capacity = n;
    67. 	php->size = n;
    68. 	//移入数据
    69. 	memcpy(php->a, a, sizeof(HPDataType) * n);
    70. 	//调整建堆	
    71. 	//向下调整
    72. 	for (int i = (n - 1 - 1) / 2; i >= 0; i--)
    73. 	{
    74. 		AdjustDown(php->a, n, i);
    75. 	}
    76. }
    77.  
    78. //销毁
    79. void HeapDestroy(HP* php)
    80. {
    81. 	assert(php);
    82. 	free(php->a);
    83. 	php->a = NULL;
    84. 	php->capacity = 0;
    85. 	php->size = 0;
    86. }
    87.  
    88. int main()
    89. {
    90. 	HPDataType a[] = { 2,4,3,5,8,5,1,6,9,10,23,45,17,19 };
    91. 	HP hp;
    92. 	HeapInitArray(&hp, a, sizeof(a) / sizeof(a[0]));
    93. 	HeapPrint(&hp);
    94.  
    95. 	HeapDestroy(&hp);
    96. 	return 0;
    97. }

    运行结果：

    可以发现转换为二叉树形式后是一个小堆结构。

    2、向上调整（时间复杂度：O(NlogN)、空间复杂度：O(1)（不计算堆所占的空间））：因为向上调整时，要保证前边的数据是堆，因此我们可以从数组的第二个元素开始，逐个元素向上调整，一直调整到最后一个叶子结点。

    代码如下：

    1. typedef int HPDataType;
    2. typedef struct Heap
    3. {
    4. 	HPDataType* a;
    5. 	int size;
    6. 	int capacity;
    7. }HP;
    8.  
    9. //打印
    10. void HeapPrint(HP* php)
    11. {
    12. 	assert(php);
    13. 	int i = 0;
    14. 	for (i = 0; i < php->size; i++)
    15. 	{
    16. 		printf("%d ", php->a[i]);
    17. 	}
    18. 	printf("\n");
    19. }
    20.  
    21. //交换函数
    22. void Swap(HPDataType* p1, HPDataType* p2)
    23. {
    24. 	HPDataType k = *p1;
    25. 	*p1 = *p2;
    26. 	*p2 = k;
    27. }
    28.  
    29. //向上调整(小堆)
    30. void AdjustUp(HPDataType* a, int child)
    31. {
    32. 	assert(a);
    33. 	int parent = (child - 1) / 2;
    34. 	while (child > 0)
    35. 	{
    36. 		if (a[child] < a[parent])
    37. 		{
    38. 			Swap(&a[child], &a[parent]);
    39. 			child = parent;
    40. 			parent = (child - 1) / 2;
    41. 		}
    42. 		else
    43. 		{
    44. 			break;
    45. 		}
    46. 	}
    47. }
    48.  
    49. //初始化
    50. void HeapInitArray(HP* php, HPDataType* a, int n)
    51. {
    52. 	assert(php);
    53. 	assert(a);
    54. 	php->a = (HPDataType*)malloc(sizeof(HPDataType) * n);
    55. 	if (php->a == NULL)
    56. 	{
    57. 		perror("malloc failed");
    58. 		exit(-1);
    59. 	}
    60. 	php->capacity = n;
    61. 	php->size = n;
    62. 	//移入数据
    63. 	memcpy(php->a, a, sizeof(HPDataType) * n);
    64. 	//调整建堆	
    65. 	//向上调整
    66. 	int i = 0;
    67. 	for (i = 1; i < n; i++)
    68. 	{
    69. 		AdjustUp(php->a, i);
    70. 	}
    71. }
    72.  
    73. //销毁
    74. void HeapDestroy(HP* php)
    75. {
    76. 	assert(php);
    77. 	free(php->a);
    78. 	php->a = NULL;
    79. 	php->capacity = 0;
    80. 	php->size = 0;
    81. }
    82.  
    83. int main()
    84. {
    85. 	HPDataType a[] = { 2,4,3,5,8,5,1,6,9,10,23,45,17,19 };
    86. 	HP hp;
    87. 	HeapInitArray(&hp, a, sizeof(a) / sizeof(a[0]));
    88. 	HeapPrint(&hp);
    89.  
    90. 	HeapDestroy(&hp);
    91. 	return 0;
    92. }

    运行结果：

    与向下调整的运行结果相同，因此该组数也是一个小堆结构。

    堆的插入：

    想要在一个堆中插入数据，只需将想要插入的数据插入到堆尾，然后再对其进行向上调整，直至满足堆的结构即可。

    代码如下：

    1. typedef int HPDataType;
    2. typedef struct Heap
    3. {
    4. 	HPDataType* a;
    5. 	int size;
    6. 	int capacity;
    7. }HP;
    8.  
    9. void HeapPrint(HP* php)
    10. {
    11. 	assert(php);
    12. 	int i = 0;
    13. 	for (i = 0; i < php->size; i++)
    14. 	{
    15. 		printf("%d ", php->a[i]);
    16. 	}
    17. 	printf("\n");
    18. }
    19.  
    20. void AdjustUp(HPDataType* a, int child)
    21. {
    22. 	assert(a);
    23. 	int parent = (child - 1) / 2;
    24. 	while (child > 0)
    25. 	{
    26. 		if (a[child] < a[parent])
    27. 		{
    28. 			Swap(&a[child], &a[parent]);
    29. 			child = parent;
    30. 			parent = (child - 1) / 2;
    31. 		}
    32. 		else
    33. 		{
    34. 			break;
    35. 		}
    36. 	}
    37. }
    38.  
    39. void HeapInitArray(HP* php, HPDataType* a, int n)
    40. {
    41. 	assert(php);
    42. 	assert(a);
    43. 	php->a = (HPDataType*)malloc(sizeof(HPDataType) * n);
    44. 	if (php->a == NULL)
    45. 	{
    46. 		perror("malloc failed");
    47. 		exit(-1);
    48. 	}
    49. 	php->capacity = n;
    50. 	php->size = n;
    51. 	//移入数据
    52. 	memcpy(php->a, a, sizeof(HPDataType) * n);
    53. 	//调整建堆
    54. 	 
    55. 	//向上调整
    56. 	int i = 0;
    57. 	for (i = 1; i < n; i++)
    58. 	{
    59. 		AdjustUp(php->a, i);
    60. 	}
    61. }
    62.  
    63.  
    64. void HeapPush(HP* php, HPDataType x)
    65. {
    66. 	assert(php);
    67. 	if (php->capacity == php->size)//扩容
    68. 	{
    69. 		int newcapacity = php->capacity == 0 ? 4 : php->capacity * 2;
    70. 		HPDataType* tem = (HPDataType*)realloc(php->a, sizeof(HPDataType) * newcapacity);//当php->a为空时，realloc将执行malloc的功能
    71. 		if (tem == NULL)
    72. 		{
    73. 			perror("malloc failed");
    74. 			exit(-1);
    75. 		}
    76. 		php->capacity = newcapacity;
    77. 		php->a = tem;
    78. 	}
    79.  
    80. 	php->a[php->size] = x;
    81. 	AdjustUp(php->a, php->size);
    82. 	php->size++;
    83. }
    84.  
    85. int main()
    86. {
    87. 	HPDataType a[] = { 2,4,3,5,8,5,1,6,9,10,23,45,17,19 };
    88. 	HP hp;
    89. 	HeapInitArray(&hp, a, sizeof(a) / sizeof(a[0]));
    90. 	HeapPrint(&hp);
    91.  
    92. 	HeapPush(&hp, 0);
    93. 	HeapPrint(&hp);
    94.  
    95. 	HeapDestroy(&hp);
    96. 	return 0;
    97. }

    运行结果：

    可以发现插入0前是小堆，插入0后还是小堆，没有改变数据的存储结构。

    堆的删除：

    一般说删除堆，都是要求删除堆顶元素即根节点，要达到此目的，只需将堆顶元素与堆尾元素互换，然后删除堆尾，再对新的堆顶元素进行向下调整，直至满足堆结构即可。

    代码如下：

    1. typedef int HPDataType;
    2. typedef struct Heap
    3. {
    4. 	HPDataType* a;
    5. 	int size;
    6. 	int capacity;
    7. }HP;
    8.  
    9. void HeapPrint(HP* php)
    10. {
    11. 	assert(php);
    12. 	int i = 0;
    13. 	for (i = 0; i < php->size; i++)
    14. 	{
    15. 		printf("%d ", php->a[i]);
    16. 	}
    17. 	printf("\n");
    18. }
    19.  
    20. void AdjustDown(HPDataType* a, int n, int parent)
    21. {
    22. 	assert(a);
    23. 	int child = parent * 2 + 1;
    24. 	while (child < n)
    25. 	{
    26. 		//找出小的
    27. 		if (child + 1 < n && a[child] > a[child + 1])
    28. 		{
    29. 			child += 1;
    30. 		}
    31.  
    32. 		if (a[child] < a[parent])
    33. 		{
    34. 			Swap(&a[child], &a[parent]);
    35. 			parent = child;
    36. 			child = parent * 2 + 1;
    37. 		}
    38. 		else
    39. 		{
    40. 			break;
    41. 		}
    42. 	}
    43. }
    44.  
    45. void HeapInitArray(HP* php, HPDataType* a, int n)
    46. {
    47. 	assert(php);
    48. 	assert(a);
    49. 	php->a = (HPDataType*)malloc(sizeof(HPDataType) * n);
    50. 	if (php->a == NULL)
    51. 	{
    52. 		perror("malloc failed");
    53. 		exit(-1);
    54. 	}
    55. 	php->capacity = n;
    56. 	php->size = n;
    57. 	//移入数据
    58. 	memcpy(php->a, a, sizeof(HPDataType) * n);
    59. 	//调整建堆
    60. 	int i = 0;
    61. 	for (i = (n - 1 - 1) / 2; i >= 0; i--)
    62. 	{
    63. 		AdjustDown(php->a, n, i);
    64. 	}
    65. }
    66.  
    67. //交换函数
    68. void Swap(HPDataType* p1, HPDataType* p2)
    69. {
    70. 	HPDataType k = *p1;
    71. 	*p1 = *p2;
    72. 	*p2 = k;
    73. }
    74.  
    75. void HeapPop(HP* php)
    76. {
    77. 	assert(php);
    78. 	assert(php->size > 0);
    79. 	Swap(&php->a[0], &php->a[php->size-1]);
    80. 	php->size--;
    81. 	//向下调整
    82. 	AdjustDown(php->a, php->size, 0);
    83. }
    84.  
    85. void HeapDestroy(HP* php)
    86. {
    87. 	assert(php);
    88. 	free(php->a);
    89. 	php->a = NULL;
    90. 	php->capacity = 0;
    91. 	php->size = 0;
    92. }
    93.  
    94. int main()
    95. {
    96. 	HPDataType a[] = { 2,4,3,5,8,5,1,6,9,10,23,45,17,19 };
    97. 	HP hp;
    98. 	HeapInitArray(&hp, a, sizeof(a) / sizeof(a[0]));
    99. 	HeapPrint(&hp);
    100.  
    101. 	HeapPop(&hp);
    102. 	HeapPrint(&hp);
    103.  
    104. 	HeapDestroy(&hp);
    105. 	return 0;
    106. }

    运行结果：

    可以发现删除堆顶后并没有改变其数据的存储结构。

    堆的应用

    堆排序：

    堆排序即利用堆的思想来进行排序，思路如下：
    1. 建堆
    升序：建大堆
    降序：建小堆
    2. 利用堆删除思想来进行排序
    因为堆顶元素一定是最大值（或最小值）每次把堆顶元素与最后一个元素交换，然后把数组尾指针向前移动1，再对新的堆顶元素进行向下调整，重复上述操作，直至数组尾指针指向第一个元素，此时的数组中的数据就是一个有序的序列。

    代码如下：

    1. typedef int HPDataType;
    2. typedef struct Heap
    3. {
    4. 	HPDataType* a;
    5. 	int size;
    6. 	int capacity;
    7. }HP;
    8.  
    9. //获取根数据
    10. HPDataType HeapTop(HP* php)
    11. {
    12. 	assert(php);
    13. 	assert(php->size > 0);
    14. 	return php->a[0];
    15. }
    16. //判空
    17. bool HeapEmpty(HP* php)
    18. {
    19. 	assert(php);
    20. 	return php->size == 0;
    21. }
    22.  
    23. //交换函数
    24. void Swap(HPDataType* p1, HPDataType* p2)
    25. {
    26. 	HPDataType k = *p1;
    27. 	*p1 = *p2;
    28. 	*p2 = k;
    29. }
    30.  
    31. //向下调整(小堆)
    32. void AdjustDown(HPDataType* a, int n, int parent)
    33. {
    34. 	assert(a);
    35. 	int child = parent * 2 + 1;
    36. 	while (child < n)
    37. 	{
    38. 		//找出小的
    39. 		if (child + 1 < n && a[child] > a[child + 1])
    40. 		{
    41. 			child += 1;
    42. 		}
    43.  
    44. 		if (a[child] < a[parent])
    45. 		{
    46. 			Swap(&a[child], &a[parent]);
    47. 			parent = child;
    48. 			child = parent * 2 + 1;
    49. 		}
    50. 		else
    51. 		{
    52. 			break;
    53. 		}
    54. 	}
    55. }
    56.  
    57.  
    58. //升序
    59. //void HeapSort(int* a, int n)
    60. //{
    61. //	HP hp;
    62. //	HeapInit(&hp);
    63. //	//建堆（小堆）
    64. //	for (int i = 0; i < n; i++)
    65. //	{
    66. //		HeapPush(&hp, a[i]);
    67. //	}
    68. //	
    69. //	int i = 0;
    70. //	while (!HeapEmpty(&hp))
    71. //	{
    72. //		a[i++] = HeapTop(&hp);//每次取堆顶元素依次放入数组
    73. //		HeapPop(&hp);
    74. //	}
    75. //
    76. //	HeapDestroy(&hp);
    77. //}
    78. // 这种写法的缺点：
    79. // 1、先要创建一个堆的结构
    80. // 2、空间复杂度、时间复杂度的消耗
    81.  
    82. //降序
    83. void HeapSort(int* a, int n)
    84. {
    85. 	//原地向下调整建堆（小堆）
    86. 	for (int i = (n - 1 - 1) / 2; i >= 0; i--)
    87. 	{
    88. 		AdjustDown(a, n, i);
    89. 	}
    90.  
    91.     //排序
    92. 	int end = n - 1;
    93. 	while (end > 0)
    94. 	{
    95. 		Swap(&a[0], &a[end]);
    96. 		AdjustDown(a, end, 0);
    97. 		end--;
    98. 	}
    99. }
    100.  
    101. int main()
    102. {
    103. 	int a[] = { 2,4,3,5,8,5,1,6,9,10,23,45,17,19 };
    104. 	HeapSort(a, sizeof(a) / sizeof(int));
    105. 	int i = 0;
    106. 	for (i = 0; i < sizeof(a) / sizeof(int); i++)
    107. 	{
    108. 		printf("%d ", a[i]);
    109. 	}
    110. 	return 0;
    111. }

    运行结果：

    TOP-K问题：

    即求一组数据中前K个最大的元素或者最小的元素。

    对于Top-K问题，能想到的最简单直接的方式就是排序，然后取前k个元素（或后k个元素），但是：如果数据量非常大，排序就不太可取了(可能数据都不能一下子全部加载到内存中)。最佳的方式就是用堆的特点来解决，基本思路如下：
    1. 用数据集合中前K个元素建堆
    若求前k个最大的元素，则建小堆
    若求前k个最小的元素，则建大堆
    2. 用剩余的N-K个元素依次与堆顶元素比较，不满足则替换堆顶元素，将剩余N-K个元素依次与堆顶元素比完之后，堆中剩余的K个元素就是所求的前K个最小或者最大的元素。

    注意：当找到前k个元素后，若不单独进行排序操作，最后输出的结果并不一定是有序的！！！

    代码如下：

    1. //交换函数
    2. void Swap(int* p1, int* p2)
    3. {
    4. 	int k = *p1;
    5. 	*p1 = *p2;
    6. 	*p2 = k;
    7. }
    8.  
    9. //向下调整(小堆)
    10. void AdjustDown(int* a, int n, int parent)
    11. {
    12. 	assert(a);
    13. 	int child = parent * 2 + 1;
    14. 	while (child < n)
    15. 	{
    16. 		//找出小的
    17. 		if (child + 1 < n && a[child] > a[child + 1])
    18. 		{
    19. 			child += 1;
    20. 		}
    21.  
    22. 		if (a[child] < a[parent])
    23. 		{
    24. 			Swap(&a[child], &a[parent]);
    25. 			parent = child;
    26. 			child = parent * 2 + 1;
    27. 		}
    28. 		else
    29. 		{
    30. 			break;
    31. 		}
    32. 	}
    33. }
    34.  
    35. //降序
    36. void HeapSort(int* a, int n)
    37. {
    38. 	//原地向下调整建堆（小堆）
    39. 	for (int i = (n - 1 - 1) / 2; i >= 0; i--)
    40. 	{
    41. 		AdjustDown(a, n, i);
    42. 	}
    43.  
    44.     //排序
    45. 	int end = n - 1;
    46. 	while (end > 0)
    47. 	{
    48. 		Swap(&a[0], &a[end]);
    49. 		AdjustDown(a, end, 0);
    50. 		end--;
    51. 	}
    52. }
    53.  
    54.  
    55. //前k个最大的
    56. void PrintTopK(int * a, int k, int n)
    57. {
    58. 	//建堆  前k个元素建堆(小堆)
    59. 	for (int i = (k - 2) / 2; i >= 0; --i)
    60. 	{
    61. 		AdjustDown(a, k, i);
    62. 	}
    63.  
    64. 	//将剩余n-k个元素依次与堆顶元素比较
    65. 	int i = k;
    66. 	while (i
    67. 	{
    68. 		if (a[i]>a[0])
    69. 		{
    70. 			//替换
    71. 			a[0] = a[i];
    72. 			//向下调整
    73. 			AdjustDown(a, k, 0);
    74. 		}
    75. 		i++;
    76. 	}
    77.  
    78. 	for (int i = 0; i < k; i++)
    79. 	{
    80. 		printf("%d ", a[i]);
    81. 	}
    82. 	printf("\n");
    83. }
    84.  
    85. int main()
    86. {
    87. 	int a1[] = { 2,4,3,5,8,5,1,6,9,10,23,45,17,19 };
    88. 	int n = sizeof(a1) / sizeof(a1[0]);
    89. 	HeapSort(a1, n);
    90. 	int i = 0;
    91. 	for (i = 0; i < n; i++)
    92. 	{
    93. 		printf("%d ", a1[i]);
    94. 	}
    95. 	printf("\n");
    96.  
    97. 	int a2[] = { 2,4,3,5,8,5,1,6,9,10,23,45,17,19 };
    98. 	PrintTopK(a2, 5, n);
    99. 	return 0;
    100. }

    运行结果：

    可以发现，虽然找出来最大的前 k 个元素，但是它们并不是有序的，这是因为我们在程序中找最大的前k个元素时，利用了堆结构，因此，当程序找到最大的前k个元素后，这k个元素是按照堆结构存储的，因此打印出来后并不是有序的。

    想要打印出来的结果也是有序的，可以在找到后，再对数组前k个元素进行一次排序即可。

    代码如下：

    1. //交换函数
    2. void Swap(int* p1, int* p2)
    3. {
    4. 	int k = *p1;
    5. 	*p1 = *p2;
    6. 	*p2 = k;
    7. }
    8.  
    9. //向下调整(小堆)
    10. void AdjustDown(int* a, int n, int parent)
    11. {
    12. 	assert(a);
    13. 	int child = parent * 2 + 1;
    14. 	while (child < n)
    15. 	{
    16. 		//找出小的
    17. 		if (child + 1 < n && a[child] > a[child + 1])
    18. 		{
    19. 			child += 1;
    20. 		}
    21.  
    22. 		if (a[child] < a[parent])
    23. 		{
    24. 			Swap(&a[child], &a[parent]);
    25. 			parent = child;
    26. 			child = parent * 2 + 1;
    27. 		}
    28. 		else
    29. 		{
    30. 			break;
    31. 		}
    32. 	}
    33. }
    34.  
    35. //降序
    36. void HeapSort(int* a, int n)
    37. {
    38. 	//原地向下调整建堆（小堆）
    39. 	for (int i = (n - 1 - 1) / 2; i >= 0; i--)
    40. 	{
    41. 		AdjustDown(a, n, i);
    42. 	}
    43.  
    44.     //排序
    45. 	int end = n - 1;
    46. 	while (end > 0)
    47. 	{
    48. 		Swap(&a[0], &a[end]);
    49. 		AdjustDown(a, end, 0);
    50. 		end--;
    51. 	}
    52. }
    53.  
    54.  
    55. //前k个最大的
    56. void PrintTopK(int * a, int k, int n)
    57. {
    58. 	//建堆  前k个元素建堆(小堆)
    59. 	for (int i = (k - 2) / 2; i >= 0; --i)
    60. 	{
    61. 		AdjustDown(a, k, i);
    62. 	}
    63.  
    64. 	//将剩余n-k个元素依次与堆顶元素比较
    65. 	int i = k;
    66. 	while (i
    67. 	{
    68. 		if (a[i]>a[0])
    69. 		{
    70. 			//替换
    71. 			a[0] = a[i];
    72. 			//向下调整
    73. 			AdjustDown(a, k, 0);
    74. 		}
    75. 		i++;
    76. 	}
    77.  
    78. 	for (int i = 0; i < k; i++)
    79. 	{
    80. 		printf("%d ", a[i]);
    81. 	}
    82. 	printf("\n");
    83. }
    84.  
    85. int main()
    86. {
    87. 	int a1[] = { 2,4,3,5,8,5,1,6,9,10,23,45,17,19 };
    88. 	int n = sizeof(a1) / sizeof(a1[0]);
    89. 	HeapSort(a1, n);
    90. 	int i = 0;
    91. 	for (i = 0; i < n; i++)
    92. 	{
    93. 		printf("%d ", a1[i]);
    94. 	}
    95. 	printf("\n");
    96.  
    97. 	int a2[] = { 2,4,3,5,8,5,1,6,9,10,23,45,17,19 };
    98. 	PrintTopK(a2, 5, n);
    99.     //单独对前5个元素排序
    100.     HeapSort(a2, 5);
    101. 	for (i = 0; i < 5; i++)
    102. 	{
    103. 		printf("%d ", a2[i]);
    104. 	}
    105. 	return 0;
    106. }

    运行结果：

  • 相关阅读:
    【CQF Finance Class 3 债券】
    理德外汇名人故事：“专家中的专家”约翰·内夫：教你系统的投资方法
    jQuery总结
    Java 中的全部锁
    OpenHarmony 入门——搭建OpenHarmony本地应用开发环境
    iTOP-RK3568开发板Linux 修改kernel logo
    Power BI 如何添加图标/自定义按钮
    基于Spring Boot+Vue的健身房管理系统(协同过滤算法、功能非常多)
    SSM整合
    自定义实现简易版ArrayList
  • 原文地址：https://blog.csdn.net/qq_74190188/article/details/132832264