7-5 堆中的路径 (25分)
将一系列给定数字插入一个初始为空的小顶堆H[]。随后对任意给定的下标i,打印从H[i]到根结点的路径。
输入格式:
每组测试第1行包含2个正整数N和M(≤1000),分别是插入元素的个数、以及需要打印的路径条数。下一行给出区间[-10000, 10000]内的N个要被插入一个初始为空的小顶堆的整数。最后一行给出M个下标。
输出格式:
对输入中给出的每个下标i,在一行中输出从H[i]到根结点的路径上的数据。数字间以1个空格分隔,行末不得有多余空格。
思路:采用数组建堆,将数组第一位设为哨兵,因为是最小堆,所以设为一个极小值,然后插入操作,此时堆的元素个数为size,所以插入从++size开始,比较和父结点大小,如果比父结点小,就和父结点交换位置,建好堆之后依次往上输出即可,不断/2,输出其祖先结点,直到H【1】
#include<stdio.h>
#define MAXN 1001
#define MINH -10001
int H[MAXN],size;
void Create()
{
size=0;
H[0]=MINH;//设置哨兵
}
void Insert(int x)
{
int i;
for(i=++size;H[i/2]>x;i/=2)//和父结点进行比较,如果比父节点大,就交换位置
{
H[i]=H[i/2];
}
H[i]=x;
}
int main()
{
int n,m,temp,way;
scanf("%d %d",&n,&m);//输入插入元素个数和需打印路径个数
Create();//初始化堆
for(int i=0;i<n;i++)
{
scanf("%d",&temp);
Insert(temp);
}
for(int j=0;j<m;j++)
{
scanf("%d",&way);
printf("%d",H[way]);
for(int z=way/2;z>0;z/=2)
{
printf(" %d",H[z]);
}
printf("\n");
}
}