一、树和二叉树
1. 为什么使用树?
我们常用树这种数据结构,原因是树结合了有序数组和链表的优点,这使得在树中查找数据项的速度和在有序数组中查找一样快,而且插入和删除数据项的速度也同链表一样。下面我们具体分析一下有序数组和链表的缺点,树是怎样使两者的优点集中在一起的。
- 在有序数组中插入新数据很慢
在有序数组中查找某个值我们可以通过二分查找来做,二分要求数据是有序的,首先查看数组的中间项的值,如果待查数据大于这个中间项的值,就在数组的后半段查找,否则在前半段查找,反复进行这个过程、不断缩小查找范围,最终查找的时间复杂度为O(logN)。
假如现在要向有序数组中插入一个新数据项,那么首先要找到新数据项插入的位置,然后把所有在其后的数据项都后移一位,删除时也是如此。这样的多次移动是很费时的,平均来说要移动数组中一半的数据项(移动N/2次)。
由此可见,如要做多次插入删除操作,有序数组这种数据结构显然是不行的。 - 在链表中查找数据很慢
链表各元素之间的连接是依靠指针的,这使得链表的插入删除操作都很快,只需改变指针的指向(引用)就可以了,这些操作的时间复杂度为O(1)。
可是在链表中进行查找就很麻烦了,对链表的访问必须从头开始,依次访问链表中的每个数据项直至找到。平均需要访问N/2个数据项,并依次进行比较,这个过程的时间复杂度为O(N)。
那么是否有一种数据结构,既能像链表那样进行快速的插入删除数据,又能像有序数组那样快速查找呢?树就实现了这一点。
2. 什么是树
这里的树和生活中的树类似,具有根、树枝、叶子等结构。简单来说,树由边连接的节点构成。在数据结构中,树的定义如下:
树(Tree)是n(n>=0)个节点的有限集。当n=0时称为空树。在任意一个非空树中都有如下特点:
1. 有且仅有一个特点的节点,称为根。
2. 当n>1时,其余节点可分为m(m>0)个互不相交的有限集,每一个集合本身又是一个树,称为根的子树。
下图展示了一个树的例子,在图中我们使用圆来代表节点,连接圆的直线代表边。
这里介绍一下与树相关的术语:
- 路径:从一个节点开始沿着节点的边走到另一个节点,所经过的节点的序列就是路径。如1-2-4-8。
- 根:树顶端的节点称为“根”。一棵树只有一个根。如果要把一个节点和边的几个定义为树,那么从根到其他任何一个节点都必须有且仅有一条路径。
- 父节点:每个节点(除根外)都恰好有一条边向上连接到另一个节点,上面的节点就称为下面节点的“父节点”。
- 子节点:每个节点都可能有一条或多条边向下连接其他节点,下面的这些节点就被称为“子节点”。
- 叶节点:没有子节点的节点称为“叶子节点”或简称“叶节点”。一个树中可以有很多叶子节点。
- 子树:每个节点都可以作为”子树“的根,他和它所有的子节点,子节点的子节点等都包含在子树中。
- 层:一个节点的层数是指从根开始到这个节点有多少”代“。假设根是第0层,它的子节点就是第1层,以此类推。
3.一种典型的树结构——二叉树
二叉树(binary tree)是树的一种特殊形式。这种树的每个节点最多只能有两个子节点。(注意是最多,可以是一个,也可以没有子节点)
二叉树节点的两个子节点(孩子节点),一个被称为左孩子(left child),一个被称为右孩子(right child)。这两个子节点的顺序是固定的,不能颠倒或混淆。
二叉树本身还存在两种特殊的形式,一个是满二叉树、一个是完全二叉树。
-
满二叉树:
一个二叉树的所有非叶子节点都存在左右孩子,并且所有叶子节点都在同一个层级上,那么这个树就是满二叉树。简单说,满二叉树的每一个分支都是满的。
下图就是一个满二叉树。
-
完全二叉树:
对一个又n个节点的二叉树,按层级顺序编号,则所有节点的编号为从1到n。如果这个树所有节点和同样深度的满二叉树的编号为从1到n的节点位置相同,则这个二叉树为完全二叉树。
下图就是一个完全二叉树,二叉树编号1-4的4个节点,和前面满二叉树编号从1到7的位置对应,因此这是一个完全二叉树。
可以这样理解,在满二叉树中去掉一些子节点及其下面连接的子节点就得到了完全二叉树。
总结: 满二叉树要求所有分支都是满的;而完全二叉树只需保证最后一个节点之前开始的节点都齐全即可。
下面主要介绍二叉树的存储结构和应用。
4.二叉树的存储结构
二叉树是一种”逻辑结构“,可以通过多种物理结构来表达。这里以链式存储结构和数组这两种物理结构为例来思考二叉树的存储结构。
(1)链式存储结构
这是二叉树最直观的存储方式。链表是“一对一”的存储方式,没一个链表节点拥有data变量和一个指向下一节点的next指针。
而再二叉树中,我们需要又两个指针来分别指向一个节点的左孩子和右孩子节点。每一个结点由下图的三部分构成:
- 存储数据的data变量
- 指向左孩子的left指针
- 指向右孩子的right指针
(2)数组
在使用数组存储时,会按照一定的层级顺序把各个结点都放置到数组中对应的位置上。如果某一个结点的左孩子/右孩子不存在,数组上也要把这一位空出来。
为了方便的在数组中定位二叉树的孩子节点和父节点,这里我们按下面的规则来进行对应。
如果一个父节点的下标是parent,那么它的左孩子节点的下标是2*parent+1;右孩子节点的下标是2*parent+2。反之,一个左孩子节点的下标是left,则他的父节点下标就是(left-1)/2。
一个例子如下图,以节点7为例,它的父节点是5,是5的右孩子,而5存在数组的4号位置,因此节点7的位置是4*2+2=10。
从上面的图例中我们不难发现,对于一个稀疏的二叉树来说,采用数组来表示是很浪费空间的。而完全二叉树比较适合用这种方式来存储(如二叉堆)。
5.二叉树的应用
(1)查找(二叉查找树)
二叉树的树形结构十分适合作索引。而二叉查找树(binary search tree)的主要作用就是进行查找。
一个标准的二叉查找树如下图所示,二叉查找树在普通二叉树的基础上增加了下面三个条件:
- 如果左子树不为空,则左子树上的所有节点的值均小于根节点的值。
- 如果右子树不为空,则右子树上的所有节点的值均大于根节点的值。
- 左、右子树也都为二叉查找树。
这样的要求能大大方便我们进行查找。
下面以查找值为4的节点为例,来看一下如何完成查找操作。
- 访问根节点6,发现4<6,应该前往左孩子节点。
- 访问节点6的左孩子节点3,发现4>3,应该前往右孩子节点。
- 访问节点3的右孩子节点4,发现4=4,找到。
总结:对于一个节点分布相对均衡的二叉查找树来说,如节点总数为n,那么搜索节点的时间复杂度为O(nlogn)。这种依靠比较大小来逐步查找的方式,类似二分查找。
(2)有序插入(二叉排序树)
在二叉查找树中我们要求左子树小于父节点,右子树大于父节点,因此保证了二叉树具有一定范围内的有序。二叉查找树也称二叉排序树(binary sort tree)。新插入二叉排序树的节点也要遵守这个规则来插入。
例如现在要插入一个节点5,先和根比较,发现5<6,因此应该去找左孩子;左孩子3<5,因此应该去找右孩子;右孩子4<5,因此应该去找右孩子;4没有右孩子,因此最终放在下图这个位置上。
下面来思考这样一个情况,在下图这个二叉排序树中插入4、3、2、1这四个节点,这时我们发现产生了一个”失衡“的二叉树,这会造成深度过深,查询节点的时间复杂度也退化到O(n)。这时就需要我们对二叉树进行平衡,二叉树的自平衡方式有红黑树、AVL树、树堆等。
二、二叉树的遍历
二叉树和数组不同,是非线性的数据结构,因此在遍历时要按照一定的规则来遍历,不同的规则遍历输出的序列顺序也不同。
从节点之间的位置关系的角度看,二叉树的遍历分为以下四种:
- 前序遍历
- 中序遍历
- 后序遍历
- 层序遍历
从宏观角度看,二叉树的遍历可归结为两类 - 深度优先遍历(前、中、后序遍历)。
- 广度优先遍历(层序遍历)。
1.深度优先遍历
深度优先,就是偏向纵深,”刨根问底“的访问节点。下面分别来看三种深度优先遍历方式。
(1)前序遍历
输出顺序:根节点->左子树->右子树。(即根节点在前)
以下图为例进行前序遍历。
-
首先输出根节点1。
-
由于节点1存在左孩子,则输出左孩子节点2。
-
由于节点2存在左孩子,则输出左孩子节点4。
-
节点4没有左孩子、右孩子,因此回到它的父节点2,发现2有右孩子,输出右孩子节点5。
-
节点5没有左孩子、右孩子,因此回到根节点1,节点1有右孩子,输出右孩子节点3。
-
节点3没有左孩子,有右孩子,因此输出右孩子节点6。
至此,所有节点都已经遍历完。
(2)中序遍历
输出顺序:左子树->根节点->右子树。(即根节点在中间)
还是用同样的例子来说明中序遍历的过程。
-
首先访问根节点的左孩子,如果这个左孩子还有左孩子就继续找下去,直到不再有左孩子为止。找到没有左孩子的节点是4,输出。
-
此时向上找这个节点的父节点2,输出。
-
找到这个节点的右孩子5,输出。
-
输出根节点1。
-
根节点的右孩子3没有左节点,因此输出它自己。
-
最后输出右节点6。
(2)后序遍历
输出顺序:左子树->右子树->根节点。(根节点最后输出)
还是以下图来说明后序遍历的过程。
- 选择没有左孩子的第一个节点输出,4被输出。
- 然后输出右节点5。
- 左右孩子都输出了,最后输出父节点2。
- 输出不再有右孩子的右节点6。
- 最后输出它的父节点3。
- 最终输出根节点1,遍历结束。
2.深度优先遍历的代码实现
对二叉树进行深度优先遍历可以采用递归和非递归两种方式,递归比较简单但是理解上可能不太容易。首先看下面对二叉树类的构建。
public class TreeNode {
int data;//数据
TreeNode leftChild;//左孩子
TreeNode rightChild;//右孩子
TreeNode(int data){
this.data=data;
}
}
然后我们首先要对二叉树进行构造,构造一个二叉树然后才能对它进行遍历,我们构建下图这样一个二叉树,使用前序的顺序输入如下数据:
3,2,9,null,null,10,null,null,8,null,4
我们先使用代码简单的递归来实现遍历,完整代码如下:
public class Main {
public static void main(String[] args) {
//固定生成一个二叉树序列 顺序是前序插入的
LinkedList<Integer> inputList=new LinkedList<Integer>(Arrays.asList(3,2,9,null,null,10,null,null,8,null,4));
TreeNode treeNode=createTree(inputList);
System.out.println("前序遍历:");preOrderTraveral(treeNode);
System.out.println("中序遍历:");inOrderTraveral(treeNode);
System.out.println("后序遍历:");postOrderTraveral(treeNode);
}
/*
创建一个二叉树
*/
public static TreeNode createTree(LinkedList<Integer> inputList){
TreeNode node=null;
if(inputList==null||inputList.isEmpty()){
return null;
}
Integer data=inputList.removeFirst();//取第一项并在原list里删除
if(data!=null){
node=new TreeNode(data);//new一个新节点
node.leftChild=createTree(inputList);//继续创建左孩子
node.rightChild=createTree(inputList);//继续创建右孩子
}
return node;
}
/*
二叉树前序遍历 根节点->左孩子->右孩子
*/
public static void preOrderTraveral(TreeNode node){
if(node==null){
return;
}
System.out.println(node.data);//输出当前节点数据
preOrderTraveral(node.leftChild);//递归输出当前节点的左孩子遍历
preOrderTraveral(node.rightChild);//递归输出当前节点的右孩子遍历
}
/*
二叉树中序遍历 左孩子->根节点->右孩子
*/
public static void inOrderTraveral(TreeNode node){
if(node==null){
return;
}
inOrderTraveral(node.leftChild);//递归输出当前节点的左孩子遍历
System.out.println(node.data);//输出当前节点数据
inOrderTraveral(node.rightChild);//递归输出当前节点的右孩子遍历
}
/*
二叉树后序遍历 左孩子->右孩子->根节点
*/
public static void postOrderTraveral(TreeNode node){
if(node==null){
return;
}
postOrderTraveral(node.leftChild);//递归输出当前节点的左孩子遍历
postOrderTraveral(node.rightChild);//递归输出当前节点的右孩子遍历
System.out.println(node.data);//输出当前节点数据
}
}
测试运行结果如下:
前序遍历:
3
2
9
10
8
4
中序遍历:
9
2
10
3
8
4
后序遍历:
9
10
2
4
8
3
Process finished with exit code 0
当然也可以用非递归方式来实现遍历,这里借助栈来实现同样的操作。
下面以前序遍历为例,说明具体实现过程。
- 首先将根节点1入栈。
- 然后选择左孩子2,入栈。
- 然后选择2的左孩子4,入栈。
- 此时发现节点4没有左孩子和右孩子,需要回溯到节点2继续。这时我们将栈顶元素4出栈,就又找到了节点2,发现节点2有右孩子5。
此时节点2的左右孩子节点都被找到了,因此也从栈中弹出,然后将5入栈。
- 节点5没有左孩子、右孩子,同样需要回溯到节点1。我们让节点5出栈。然后发现节点1有右孩子3,节点1的左孩子右孩子都被找到了,节点1出栈,节点3入栈。
- 节点3的右孩子是节点6,没有左孩子,这样节点3的左右孩子也都被发现了,节点3出栈,节点6入栈。
- 节点6没有左右孩子,因此它出栈。这时栈为空,遍历结束。
这个过程的代码实现如下:
/*
二叉树非递归前序遍历
*/
public static void preOrderTraveralWithStack(TreeNode treeNode){
Stack<TreeNode> stack=new Stack<TreeNode>();//初始化一个栈
while (treeNode!=null||!stack.isEmpty()){//还存在数据就继续
while (treeNode!=null){//节点存在
System.out.println(treeNode.data);
stack.push(treeNode);//该节点入栈
treeNode=treeNode.leftChild;//迭代访问左孩子节点
}
if(!stack.isEmpty()){//栈不为空,且节点没有左孩子了
treeNode=stack.pop();//弹出栈顶节点
treeNode=treeNode.rightChild;//迭代访问右孩子节点
}
}
}
3.广度优先遍历
广度优先遍历和在一个方向上”直接到底“的深度优先遍历相反,它是先在各个方向上各走出一步,再在各个方向上走出第2、3步…直到各个方向全部走完。
下面以二叉树的层序遍历为例说明广度优先遍历的特点。
层序遍历,是指二叉树按照从根节点到叶子节点的层次关系,一层层地横向遍历各个节点。
我们知道,二叉树同一层次的节点之间是没有任何关联的,如何实现层序遍历呢?我们使用数据结构队列。以下图为例,一个二叉树的层序遍历实现如下:
-
根节点1插入队列。
-
节点1出队,输出节点1,并得到节点1的左孩子节点2和右孩子节点3。让节点2、3入队。
-
节点2出队,输出节点2,并得到节点2的左孩子节点4、右孩子节点5。让节点4、5入队。
-
节点3出队,输出节点3,并得到节点3的右孩子节点6,让节点6入队。
-
节点4出队,输出节点4,因为4没有子节点,故没有新节点入队。
-
节点5出队,输出节点5,因为节点5也没有子节点,没有新节点入队。
-
节点6出队,输出节点6,因为节点6没有子节点,所以没有新入队节点。此时栈为空,所有节点遍历结束。
4.广度优先遍历的代码实现
/*
二叉树的层序遍历
队列实现
*/
public static void levelOrderTraversal(TreeNode root){
Queue<TreeNode> queue=new LinkedList<TreeNode>();//声明一个队列
queue.offer(root);//加入节点
while (!queue.isEmpty()){//队列不为空
TreeNode node=queue.poll();//第一个队列中的节点出队
System.out.println(node.data);//输出这个节点
if(node.leftChild!=null){
queue.offer(node.leftChild);//左孩子存在,就把左孩子加队列
}
if(node.rightChild!=null){
queue.offer(node.rightChild);//右孩子存在,就把右孩子加队列
}
}
}
三、二叉堆
1.什么是二叉堆
二叉堆可以通过自身调整,让最大或最小的元素移动到顶点。
二叉堆是实现堆排序和优先队列的基础。
二叉堆本质上是一种完全二叉树,它分为以下两个类型。
-
最大堆。
任何一个父节点的值都大于或者等于它左、右孩子节点的值。下图就是一个最大堆。
-
最小堆。
任何一个父节点的值,都小于或等于它左、右孩子节点的值。 下图是一个最小堆。
二叉堆的根节点叫做堆顶。最大堆和最小堆的特点决定了:最大堆的堆顶是整个堆中的最大元素;最小堆的堆顶是整个堆中的最小元素。
2.堆的构建(基于二叉堆的自我调整)
对于二叉堆有如下几种操作。
- 插入节点。
- 删除节点。
- 构建二叉堆。
这三种操作都基于堆的自我调整。堆的自我调整就是把一个不符合堆的性质的完全二叉树,调整成一个堆。
下面以最小堆的构建为例,看一个完全二叉树是如何自我调整的。
(1)插入节点
在二叉堆中,新插入的位置都是完全二叉树的最后一个位置。例如现在往下图所示的二叉树中插入一个节点0。
此时新节点的父节点5比0大,需要将新节点”上浮“,和父节点进行位置交换。
继续把新节点0和它的父节点3作比较,0<3,新节点和父节点3交换位置。
继续比较,最终新节点0上浮到堆顶位置,结束。
(2)删除节点
在二叉堆中,删除节点只能删除堆顶的节点。假如现在我们要删除下面二叉树的堆顶1。
删掉堆顶后,为了保证树的结构,我们采用把堆的最后一个节点10顶替到堆顶,然后再进行自我调整的方式完成这个删除操作的”善后“。
然后开始用自我调整的方式对这个临时的堆顶10进行调整。10和左右孩子进行比较,如果左右孩子节点中最小的一个比10小,就让节点10和它互换,完成”下沉“。(2和10进行交换)
同样的再次进行比较、互换,最后节点10”下沉“到了原节点7的位置。
这样,二叉堆就完成了自我调整,又恢复到了一个正常的状态。
(3)构建二叉堆
构建二叉堆的过程,就是把一个无序的完全二叉树调整为二叉堆,本质是让所有非叶子节点依次“下沉”。
下面给出一个无序完全二叉树的例子,我们以该二叉树为例来构建一个二叉堆。
首先找到最后一个非叶子节点10。如果节点10大于它左右孩子节点中的最小的一个 ,则节点10“下沉”。
然后是倒数第二个非叶子节点3,如果节点3大于它左右孩子节点中的最小的一个 ,则节点3“下沉”。
然后找到节点1,如果节点1大于它左右孩子节点中的最小的一个 ,则节点1“下沉”。(不用改变)
然后找到节点7,如果节点7大于它左右孩子节点中的最小的一个 ,则节点7“下沉”。
节点7继续进行比较,发现还需“下沉”。
经过几次比较和“下沉”操作,最终实现了每一个节点都小于它的左右孩子节点,一个无序的完全二叉树就被构建成了一个最小堆。
拓展:
堆的插入和删除操作的时间复杂度:O(log n)
堆的构建的时间复杂度:O(n)
3.二叉堆的代码实现
需要注意的是,二叉堆虽然是一个完全二叉树,但是它的存储方式确是顺序存储。二叉堆的所有节点是存储在数组中的。
数组没有左右指针这种结构,那么靠什么来定位一个父节点的左右孩子呢?我们依靠数组下标就能完成定位。
规则是:假设父节点的下标是parent,那么它的左孩子下标就是parent2+1,右孩子下标就是parent2+2。一个图例如下:
下面是二叉堆构建的代码实例:
/*
二叉堆的实现
*/
public class Main {
public static void main(String[] args) {
int []array =new int[]{1,3,2,6,5,7,8,9,10,0};
upMove(array);
System.out.println(Arrays.toString(array));
array=new int[]{7,1,3,10,5,2,8,9,6};
buildHeap(array);
System.out.println(Arrays.toString(array));
}
/*
节点"上浮"移动 【用于在二叉堆中插入一个新节点时】
@param array 待调整的heap
*/
public static void upMove(int []array){
int childIndex=array.length-1;
int parentIndex=(childIndex-1)/2;
int tmp=array[childIndex];//保存插入的叶子节点值,赋值用
while (childIndex>0 &&tmp<array[parentIndex]){//找到的父节点还比新节点大,继续上浮
array[childIndex]=array[parentIndex];//父节点的值放到原子节点的位置
childIndex=parentIndex;//子节点下标“上浮”,上位到原来父节点的位置
parentIndex=(parentIndex-1)/2;//再向上找父节点,循环判断还要不要继续上浮了
}
array[childIndex]=tmp;//最终找到的父节点位置就是新节点要放入的位置
}
/*
节点"下沉"移动 【用于将一个完全二叉树构建成二叉堆时】
@param array 待调整的heap
@param parentIndex 要下沉的父节点
@param length heap的大小
*/
public static void downMove(int []array,int parentIndex,int length){
int childIndex=2*parentIndex+1;
int tmp=array[parentIndex];//保存父节点值,赋值用
while (childIndex<length){
//如果右孩子存在且右孩子小于左孩子的值,定位到右孩子
if(childIndex+1<length && array[childIndex+1]<array[childIndex]){
childIndex++;
}
//如果父节点比子节点值小,完成下沉,跳出
if(tmp<=array[childIndex]){
break;
}
array[parentIndex]=array[childIndex];
parentIndex=childIndex;//交换
childIndex=2*childIndex+1;//继续找下一个子节点,循环判断是否还需要下沉
}
array[parentIndex]=tmp;
}
/*
构建堆
@param array 待调整heap
*/
public static void buildHeap(int []array){
//从【最后一个非叶子节点】开始,依次做“下沉”调整
for(int i=(array.length-2)/2; i>=0; i--){
downMove(array,i,array.length);
}
}
}
测试运行结果如下:
[0, 1, 2, 6, 3, 7, 8, 9, 10, 5]
[1, 5, 2, 6, 7, 3, 8, 9, 10]
Process finished with exit code 0
四、优先队列
1.普通队列与优先队列
我们回顾一下队列的特点,队列是一种遵循先进先出(FIFO)的数据结构,在入队时,新元素将被置于队尾:
在出队时,只有队头元素能被移出:
那么优先队列和普通队列之间有何区别呢?
优先队列不再遵循FIFO规则,它的优先级规则如下:
- 最大优先队列:无论入队顺序如何,总是当前最大的元素优先出队。
- 最小优先队列:无论入队顺序如何,总是当前最小的元素优先出队。
用下图来说明这个规则,这是一个最大优先队列,最大元素是8,尽管8并不是队首元素,但出队的元素却是8。
2.优先队列的实现
通过前面的介绍,最大堆和最小堆来实现最大优先队列和最小优先队列是最佳的选择。每一次入队操作就是堆的插入操作,每一次出队操作就是删除堆顶节点。
下面以最大优先队列为例,说明入队、出队操作的过程。
(1)入队操作
- 插入新节点5。
- 新节点5“上浮”到一个合适的位置,这里就上浮一次即满足条件。
(2)出队操作
-
让原堆顶节点10出队。
-
把最后一个节点1直接替换到堆顶,顶替出队的原堆顶。
-
使节点1“下沉”到合适位置,最终节点9称为堆顶。
总结:二叉堆节点“上浮”、“下沉”的时间复杂度都是O(logn),所以优先队列入队和出队的时间复杂度也是O(logn)。
(3)代码实现
拓展:可以使用JDK提供的java.util.PriorityQueue直接完成优先队列的实现。关于PriorityQueue的介绍如下:
Java中PriorityQueue实现了Queue接口,不允许放入null元素;其通过堆实现,具体说是通过完全二叉树(complete binary tree)实现的小顶堆(任意一个非叶子节点的权值,都不大于其左右子节点的权值),也就意味着可以通过数组来作为PriorityQueue的底层实现。
优先队列的作用是能保证每次取出的元素都是队列中权值最小的(Java的优先队列每次取最小元素,C++的优先队列每次取最大元素)。元素大小的评判可以通过元素本身的自然顺序(natural ordering),也可以通过构造时传入的比较器来定义。
在如下代码中,我们构建了一个最小优先队列,即每次都是最小的元素先出队。
public class PriorityQueue {
private int []array;
private int size;
public PriorityQueue(){
array=new int[32];//初始长度32
}
/*
队列扩容
*/
private void resize(){
//容量x2
int newSize =this.size*2;
this.array= Arrays.copyOf(this.array,newSize);
}
/*
入队操作
*/
public void enQueue(int key){
if(size>=array.length){//超出最大长度,需要扩容
resize();
}
array[size++]=key;
upMove();//上浮
}
/*
出队操作
*/
public int deQueue() throws Exception{
if(size<=0){//空队列,抛出异常
throw new Exception("null queue");
}
//获取堆顶元素
int head=array[0];
//让最后一个元素移动到堆顶
array[0]=array[--size];
downMove();//下沉
return head;
}
/*
节点"上浮"移动
*/
private void upMove(){
int childIndex=size-1;
int parentIndex=(childIndex-1)/2;
int tmp=array[childIndex];//保存插入的叶子节点值,赋值用
while (childIndex>0 &&tmp<array[parentIndex]){//找到的父节点还比新节点大,继续上浮
array[childIndex]=array[parentIndex];//父节点的值放到原子节点的位置
childIndex=parentIndex;//子节点下标“上浮”,上位到原来父节点的位置
parentIndex=(parentIndex-1)/2;//再向上找父节点,循环判断还要不要继续上浮了
}
array[childIndex]=tmp;//最终找到的父节点位置就是新节点要放入的位置
}
/*
节点"下沉"移动 【用于将一个完全二叉树构建成二叉堆时】
*/
private void downMove(){
int parentIndex =0;
int childIndex=1;
int tmp=array[parentIndex];//保存父节点值,赋值用
while (childIndex<size){
//如果右孩子存在且右孩子小于左孩子的值,定位到右孩子
if(childIndex+1<size && array[childIndex+1]<array[childIndex]){
childIndex++;
}
//如果父节点比子节点值小,完成下沉,跳出
if(tmp<=array[childIndex]){
break;
}
array[parentIndex]=array[childIndex];
parentIndex=childIndex;//交换
childIndex=2*childIndex+1;//继续找下一个子节点,循环判断是否还需要下沉
}
array[parentIndex]=tmp;
}
}
测试主方法如下:
public static void main(String[] args) throws Exception {
PriorityQueue priorityQueue=new PriorityQueue();
priorityQueue.enQueue(3);
priorityQueue.enQueue(5);
priorityQueue.enQueue(10);
priorityQueue.enQueue(2);
priorityQueue.enQueue(7);
System.out.println("出队元素:"+priorityQueue.deQueue());
System.out.println("出队元素:"+priorityQueue.deQueue());
}
测试运行结果如下:
第一次2最小,所以2出队;然后3是最小元素,所以第二次3出队。
出队元素:2
出队元素:3
Process finished with exit code 0