堆中的路径
问题描述来自PTA-数据结构与算法题目集(中文)7-5 堆中的路径:
将一系列给定数字插入一个初始为空的小顶堆H[]。随后对任意给定的下标i,打印从H[i]到根结点的路径。
输入格式:
每组测试第1行包含2个正整数N和M(≤1000),分别是插入元素的个数、以及需要打印的路径条数。下一行给出区间[-10000, 10000]内的N个要被插入一个初始为空的小顶堆的整数。最后一行给出M个下标。
输出格式:
对输入中给出的每个下标i,在一行中输出从H[i]到根结点的路径上的数据。数字间以1个空格分隔,行末不得有多余空格。
输入样例:
5 3
46 23 26 24 10
5 4 3
输出样例:
24 23 10
46 23 10
26 10
小顶堆的操作参考自shuaixio-算法与数据结构-堆的基本操作C语言实现
博客中主要为思路与算法思想,完整的代码参看我的github
堆的概念与性质
堆是一棵完全二叉树,所以我们可以用数组H[]
表示堆,这样可以简化我们许多操作。
对于从0开始下标为i
的元素,我们可以得到
p a r e n t ( i ) = ( i − 1 ) / 2 l e f t c h i l d ( i ) = i ∗ 2 + 1 r i g h t c h i l d ( i ) = i ∗ 2 + 2 parent(i)=(i-1)/2 \newline leftchild(i)=i*2+1 \newline rightchild(i)=i*2+2 parent(i)=