当前位置:   article > 正文

数据结构之堆(优先级队列)

数据结构之堆(优先级队列)

前言

在上一章我们讲了二叉树,这一节我们来讲堆(优先级队列),所以想知道堆创建,可以看一下二叉树的一些简单概念。http://t.csdnimg.cn/4jUR6icon-default.png?t=N7T8http://t.csdnimg.cn/4jUR6

目录

前言

1.概念

2.优先级队列的模拟实现

2.1堆的概念

2.2堆的性质

2.3堆的存储方式

2.4堆的创建

2.4.1向下调整

1.向下调整思路

​编辑

 2.代码实现

 3.建堆的时间复杂度

2.5堆的插入

2.5.1向上调整

2.5.2堆插入代码实现:

2.6堆元素的删除

2.6.1思路 

2.6.2.代码实现

2.7获取堆的元素个数&&获取堆顶元素&&堆的打印

3.常用接口特性

3.1PriorityQueue的特性

注意:

 3.2PriorityQueue常用接口介绍


1.概念

我们知道队列是一种先进先出的数据结构,但是在某些情况下,操作的数据可能带有优先级,出队的时候,可能需要优先级较高的元素出队列。

所以,数据结构应该提供两个最基本的操作,一个是返回最高优先级对象,二是添加新的对象。这种数据结构就是优先级队列(Priority Queue)

2.优先级队列的模拟实现

2.1堆的概念

如果有一个集合K={0,k1,,k2,...,kn-1},把集合K的元素按照完全二叉树的顺序存储方式存储在一个一维数组中,并且满足:K(i)<=K(2I+1)且K(i)<=K(2i+2)  (K(i)>=K(2I+1)且K(i)>=K(2i+2) ),i=0,1,2,3... ,则叫做小堆(或大堆)。

将根节点最小的堆叫做小根堆或最小堆。

将根节点最大的堆叫做大根堆或最大堆。

2.2堆的性质

1.堆中的某个节点总是不大于或不小于其父节点的值。

2.堆总是一棵完全二叉树。

在jDK1.8中的优先级队列底层使用了堆这种数据结构,而堆其实就是就是完全二叉树的基础进行调整的。

2.3堆的存储方式

我们从堆的概念可以知道,堆是一棵完全二叉树,所以可以层序的规则采用顺序的方式存储

注意:对于非完全二叉树,不适合采用顺序方式进行存储。

原因:为了还原二叉树,空间中必须要存储空节点,就会导致空间利用率较低。

将元素存储到数组中后,我们可以根据二叉树的性质5进行还原,假设i为节点在数组中的下标,则有:

  1. 如果i为0,则i表示的节点为根节点,否则i节点的双亲节点为(i-1)/2;
  2. 如果2*i+1小于节点个数,则节点i的左孩子下标为2*i+1,否则没有左孩子;
  3. 如果2*i+2小于节点个数,则节点的右孩子下标为2*i+2,否则没有右孩子。

2.4堆的创建

2.4.1向下调整

我们拿集合{28,16,48,13,46,45,25,36,22,42}为例,如何将其创建成堆呢??

我们可以看到,此时根节点的左右子树都不满足堆的性质。所以我们需要对每个有子树的父节点进行向下调整。

1.向下调整思路

对于一棵完全二叉树,要其变成小根堆(或大根堆),我们需要满足根节点的左右子树都是小堆(大堆)。

规则1.找出父亲节点的左右节点中值较小(或较大)的节点。

           2.找出较小值(较大值)与父亲节点进行比较。

           3.小堆:若父亲节点比左右节点中的较小值大,则进行交换,再将较小值的位置给到父亲节点,再进行向下调整。当父亲节点的值小于左右节点中的较小值时,调整停止

              大堆:若父亲节点比左右节点中的较大值小,则进行交换,再将较大值的位置给到父亲节点,再进行向下调整。当父亲节点的值小于左右节点中的较小值时,调整停止

我们以创建小根堆为例:

对于上图中的二叉树,我们可以看到其左右子树并不是小堆。

所以我们需要先对其左右子树进行向下调整:

我们从根节点位置最大的一个开始,依次递归进行调整。

我们可以看到父亲节点(46)比孩子节点(42)要大,所以要进行交换。

再让父亲节点(P)走向孩子节点(C)的位置,但是由于此时父亲节点并没有孩子节点,停止调整。再让P从节点值为13的位置开始向下调整,此时由于左右节点值都大于13,满足堆的性质,不进行交换。 

依次类推:

当P走到节点值为48的位置时,再与左右孩子中的最小值进行比较,进行互换。

当P走到父亲节点(16)的位置时,进行向下调整,再让P往下走,但此时P所处的节点其满足堆的性质,不进行互换,调整停止。 

 

此时,根节点(28)的左右子树都已经满足堆的性质,现只需要对根节点进行向下调整,就可以得到一个小根堆。

 

至此,我们就得到一个小根堆。

我们如果要创建一个大根堆,思路也是与创建小根堆的思路一样,只是在交换值时,是交换孩子节点中的较大值。

 2.代码实现

对于上面我们所推的,

其小根堆为{13,16,25,22,42,45,48,36,28,46};

其大根堆为{48,46,45,36,42,28,25,13,22,26};

  1. package MyQueue;
  2. /**
  3. * Pheap类实现了大根堆数据结构。
  4. */
  5. class Pheap {
  6. public int[] elem; // 存储堆元素的数组
  7. public int useSize; // 当前堆中元素的使用大小
  8. /**
  9. * 构造函数,初始化堆数组。
  10. *
  11. * @param size 堆数组的初始大小
  12. */
  13. public Pheap(int size){
  14. this.elem=new int[size];
  15. }
  16. /**
  17. * 使用给定数组初始化堆。
  18. *
  19. * @param arr 用于初始化堆的数组
  20. */
  21. public void init(int[] arr){
  22. for(int i=0;i<arr.length;i++){
  23. this.elem[i]=arr[i];
  24. }
  25. useSize=arr.length;
  26. }
  27. /**
  28. * 交换数组中两个元素的位置。
  29. *
  30. * @param child 需要交换的子元素下标
  31. * @param parent 需要交换的父元素下标
  32. */
  33. public void swap(int child,int parent){
  34. int temp=elem[child];
  35. elem[child]=elem[parent];
  36. elem[parent]=temp;
  37. }
  38. /**
  39. * 向下调整以维护大根堆性质。
  40. *
  41. * @param parent 需要向下调整的父节点下标
  42. * @param end 堆数组的结束下标
  43. */
  44. public void sitDownBig(int parent,int end){
  45. int child=2*parent+1;
  46. while(child<end){
  47. if(child+1<end&&elem[child]<elem[child+1]){
  48. child++;
  49. }
  50. if(elem[child]>elem[parent]){
  51. swap(child,parent);
  52. parent=child;
  53. child=2*parent+1;
  54. }else{
  55. break;
  56. }
  57. }
  58. }
  59. /**
  60. * 构建大根堆。
  61. */
  62. public void createHeapBig(){
  63. for(int parent=(useSize-1-1)/2;parent>=0;parent--){
  64. sitDownBig(parent,useSize);
  65. }
  66. }
  67. /**
  68. *构建小根堆
  69. */
  70. public void creatHeapSmall(){
  71. for(int parent=(useSize-1-1)/2;parent>=0;parent--){
  72. sitDownSmall(parent,useSize);
  73. }
  74. }
  75. /**
  76. * 将指定元素下沉以维护堆的性质。该方法用于调整二叉堆,确保从指定父节点到末尾子节点的子树满足堆的性质。
  77. *
  78. * @param parent 父节点的索引
  79. * @param end 堆数组的末尾索引
  80. */
  81. public void sitDownSmall(int parent, int end) {
  82. // 计算左子节点的索引
  83. int child = 2 * parent + 1;
  84. while (child < end) {
  85. // 如果存在右子节点,并且右子节点比左子节点大,则将当前 child 指针指向右子节点
  86. if (child + 1 < end && elem[child] > elem[child + 1]) {
  87. child++;
  88. }
  89. // 如果当前 child 节点的值小于父节点的值,则交换它们,并将 parent 更新为当前 child,继续下沉调整
  90. if (elem[child] < elem[parent]) {
  91. swap(child, parent);
  92. parent = child;
  93. // 更新 child 为新的左子节点索引
  94. child = 2 * parent + 1;
  95. } else {
  96. // 如果当前 child 节点的值不小于父节点的值,说明已满足堆的性质,结束调整
  97. break;
  98. }
  99. }
  100. }
  101. }

测试一下

  1. public class Prioirtyq {
  2. public static void main(String[] args){
  3. Pheap p=new Pheap(10);
  4. int arr[]={28,16,48,13,46,45,25,36,22,42};
  5. p.init(arr);
  6. p.creatHeapSmall();
  7. Pheap p1=new Pheap(10);
  8. p1.init(arr);
  9. p1.createHeapBig();
  10. }
  11. }

可以看到,确实是所推的那样。

 3.建堆的时间复杂度

我们假设完全二叉树的高度为h,

那么,对于第一层,其结点只有一个,但是其需要向下调整h-1层。对于第二层,其节点有2^1个,每个结点需要向下调整的次数为h-2,以此类推,对于第h-1层,其拥有的节点有2^{h-2}个,但其属于倒数第二层,所以只需要向下调整1次。

那么对于一棵完全二叉树,要想将其建成一个堆,其时间复杂度就是每层的节点数*其向下调整的次数所需要花费的时间

T(n)=2^0*(h-1)+2^1*(h-2)+2^2*(h-3)+...+2^(h-2)*1               (1)式

我们不难看出,这是一个等差✖等比求和公式,我们可以用错位相减法来求出T(n),不难看出,其公比为2.

所以在(1)式左右两边同时✖2,得

2T(n)=              2^1*(h-1)+2^2*(h-2)+2^3*(h-3)+...+2^(h-1)*1     (2)式

(2)式-(1)式可得

T(n)=2^1+2^2+2^3+...+2^(h-1)+1-h

我们将1化为2^0,

T(n)=2^0+2^1+2^2+2^3+...+2^(h-1)-h

可以看出这是一个等比数列求和公式,根据求和公式Sn=a1*(1-q^n)/1-q,

T(n)=1*(1-2^h)/(1-2)-h=2^h-1-h

由二叉树的性质我们可以得到

节点数N=2^h-1

树的高度h=log2(N+1)

带入得

T(n)=N-log2(N+1)

根据大O渐进表示法

T(n)=O(N)

所以我们建堆的时间复杂度为O(N).

向下调整的时间复杂度为O(logN).

2.5堆的插入

在一个堆中,如果我们想插入一个数据,那么就在堆尾进行插入,再进行向上调整.

我们同时也需要考虑此时堆满了没

2.5.1向上调整

思路:对于插入的节点(我们称作目标节点)

         1.将目标节点与其父亲节点进行比较。

           大根堆:如果是大根堆,当父亲节点比目标节点小,那就目标节点和父亲节点进行互换后,将父亲节点的位置给到目标节点,接着继续进行向上调整。当父亲节点比目标节点大,停止向上调整。

          小根堆:当父亲节点比目标节点大,那就目标节点和父亲节点进行互换后,将父亲节点的位置给到目标节点,接着继续进行向上调整。当父亲节点比目标节点小,停止向上调整。

  我们以小根堆插入新节点为例:

我们用上述中所创建而成的小根堆,让其插入一个值为10的节点,如图

我们可以知道,新插入的节点其父亲节点是值为42的节点,明显比值为10目标节点要大,所以要进行互换,再进行向上调整。

最后我们可以得到:

 此时小根堆为{10,13,25,22,16,45,48,36,46,42}。

2.5.2堆插入代码实现:
  1. public void pushInS(int val){
  2. // 判断堆是否已满
  3. if(isFull()){
  4. elem= Arrays.copyOf(elem,elem.length*2);
  5. }
  6. //进行插入
  7. elem[useSize++]=val;
  8. //进行向上调整
  9. sitUp(useSize-1);
  10. }
  11. public void sitUp(int child){
  12. int parent=(child-1)/2;
  13. while(child>=0){
  14. if(elem[child]<elem[parent]){
  15. swap(child,parent);
  16. child=parent;
  17. parent=(child-1)/2;
  18. }else{
  19. break;
  20. }
  21. }
  22. }
  23. /**
  24. * 检查堆是否已满。
  25. *
  26. * @return 堆是否已满的布尔值
  27. */
  28. public boolean isFull(){
  29. return useSize==elem.length;
  30. }

可以看到,确定是所推断的那样。 

2.6堆元素的删除

堆元素的删除一定是删除的堆顶元素!!!

2.6.1思路 

对顶元素的删除其实也是利用到向下调整。

1.将对顶元素与队尾的元素进行互换

2.让有效个数减1

3.再来一次向下调整

2.6.2.代码实现
  1. public int Delete(){
  2. if(isEmpty()){
  3. throw new RuntimeException("堆为空");
  4. }
  5. int val=elem[0];
  6. swap(0,useSize-1);
  7. useSize--;
  8. sitDownSmall(0,useSize);
  9. return val;
  10. }

2.7获取堆的元素个数&&获取堆顶元素&&堆的打印

  1. public int size(){
  2. return useSize;
  3. }
  4. public int peek(){
  5. if(isEmpty()){
  6. throw new RuntimeException("堆为空");
  7. }
  8. return elem[0];
  9. }
  10. public void print(){
  11. for(int i=0;i<useSize;i++){
  12. System.out.print(elem[i]+" ");
  13. }
  14. System.out.println();
  15. }

3.常用接口特性

3.1PriorityQueue的特性

在java集合框架中,提供了PriorityQueue和PriorityBlockingQueue两种类型的优先级队列,但是PriorityQueue时线程不安全的,而PriorityBlockingQueue是线程安全的。

 我们在使用PriorityQueue时,需要导入相应的包

import java.util.PriorityQueue;

注意:

1.PriorityQueue中放置的元素必须要能够比较大小,不能插入无法比较大小的对象,否则会抛出 ClassCastException异常

2. 不能插入null对象,否则会抛出NullPointerException

3. 没有容量限制,可以插入任意多个元素,其内部可以自动扩容

4. 插入和删除元素的时间复杂度为O(log2N).

5. PriorityQueue底层使用了堆数据结构

6. PriorityQueue默认情况下是小堆---即每次获取到的元素都是最小的元素

 3.2PriorityQueue常用接口介绍

1.优先级队列的构造

常用的有以下几个:

 如果想要了解更多关于优先级队列,可以点击PriorityQueue (Java 平台 SE 8 ) (oracle.com)

  1. public static void main(String[] args){
  2. PriorityQueue<Integer> pq=new PriorityQueue<>();
  3. pq.offer(1);
  4. pq.offer(2);
  5. pq.offer(3);
  6. System.out.println(pq.poll());
  7. System.out.println(pq.peek());
  8. }

 扩容规则:
 如果容量小于64时,是按照oldCapacity的2倍方式扩容的

如果容量大于等于64,是按照oldCapacity的1.5倍方式扩容的

如果容量超过MAX_ARRAY_SIZE,按照MAX_ARRAY_SIZE来进行扩容。

数据结构的堆就先到这。

若有不足之处,欢迎指正~~

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/687142
推荐阅读
相关标签
  

闽ICP备14008679号