将一系列给定数字插入一个初始为空的小顶堆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
这道题最好的解法应该是直接建堆,即先将所有元素按输入顺序存储,再调整结点位置,这样时间复杂度为O(n)
但是这题并不能这样做,因为所有的测试点都是按照将所有元素一个一个插入得到的,而堆并不唯一(最小堆只要求父节点小于两个子节点,但两个子节点之间的大小并无要求),所以和我们得到的对不一样。
正确代码如下:
#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include<stdlib.h>
#define MAXSIZE 1000
#define MINDATA -10001
#define ElementType int
typedef struct HNode* Heap;
struct HNode
{
ElementType* data;
int Size;
int Capacity;
};
Heap CreateHape()
{
Heap H = (Heap)malloc(sizeof(struct HNode));
H->Capacity = MAXSIZE;
H->Size = 0;
H->data = (ElementType*)malloc(sizeof(ElementType) * (MAXSIZE + 1));
H->data[0] = MINDATA;
int i;
for (i = 1; i <= MAXSIZE; ++i)
H->data[i] = 0;
return H;
}
void Insert(Heap H, ElementType data)
{
if (H->Size == H->Capacity)
return;
else
{
int tempnode = ++H->Size;
while (data < H->data[tempnode/2])
{
H->data[tempnode] = H->data[tempnode / 2];
tempnode = tempnode / 2;
}
H->data[tempnode] = data;
}
}
void Print(Heap H, int node)
{
printf("%d", H->data[node]);
node = node / 2;
while (H->data[node]>MINDATA)
{
printf(" %d", H->data[node]);
node = node / 2;
}
printf("\n");
}
int main()
{
int n, m, i, node, data;
scanf("%d %d", &n, &m);
Heap H;
H = CreateHape();
for (i = 0; i < n; ++i)
{
scanf("%d", &data);
Insert(H, data);
}
for (i = 0; i < m; ++i)
{
scanf("%d", &node);
Print(H, node);
}
return 0;
}