题目
将一系列给定数字插入一个初始为空的小顶堆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
要求
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB
题意理解
根据输入样例可以构造的最小堆为:
三次查询:
(1)下标为5的元素到根结点的路径:24 23 10
(2)下标为4的元素到根结点的路径:46 23 10
(3)下标为3的元素到根结点的路径:26 10
堆的表示及操作
堆是一种按一定顺序组织的完全二叉树,所以就用数组表示堆。
堆是把根结点放在下标为1的地方,0是空缺的。通过这种方式可以很容易找到父结点和孩子结点的位置。
主程序
int main()
{
1、读入n和m
2、根据输入序列建堆
3、对m个要求:打印到根的路径
return 0;
}
代码
- 使用结构体来表示堆:
#include <stdio.h>
#include <stdlib.h>
typedef int ElementType;
typedef struct HeapStruct *MinHeap;
struct HeapStruct
{
ElementType *elements;
int size;
int capacity;
};
#define ERROR -1
#define MINDATA -10001
MinHeap Create(int MaxSize)
{
MinHeap H = malloc(sizeof(struct HeapStruct));
H->elements = malloc(sizeof(ElementType) * (MaxSize + 1));
H->size = 0;
H->capacity = MaxSize;
H->elements[0] = MINDATA;
return H;
}
int IsFull(MinHeap H)
{
return H->size == H->capacity;
}
void Insert(MinHeap H, ElementType item)
{
if(IsFull(H)) {
printf("Heap is full!\n");
return;
}
int i = ++H->size;
for(; H->elements[i/2] > item; i/=2) {
H->elements[i] = H->elements[i/2];
}
H->elements[i] = item;
}
int main()
{
int N, M;
scanf("%d %d", &N, &M);
MinHeap H = Create(N);
int i;
int item;
for(i = 0; i < N; i++) {
scanf("%d", &item);
Insert(H, item);
}
int j;
for(i = 0; i < M; i++) {
scanf("%d", &j);
printf("%d", H->elements[j]);
while(j > 1) {
j /= 2;
printf(" %d", H->elements[j]);
}
printf("\n");
}
return 0;
}
运行结果:
- 或者不用结构体,直接用数组来处理,代码如下:
#include <stdio.h>
#include <stdlib.h>
#define MINDATA -10001
#define MAXN 1001
int H[MAXN];
int size;
void Create()
{
size = 0;
H[0] = MINDATA; //设置“岗哨”
}
int IsFull()
{
return size == MAXN;
}
void Insert(int item)
{
if(IsFull(H)) {
printf("Heap is full!\n");
return;
}
int i;
for(i = ++size; H[i/2] > item; i/=2) { //因为有“岗哨”,所以不用判断下标是否越界,且是从最后一个位置开始比较
H[i] = H[i/2];
}
H[i] = item;
}
int main()
{
int N, M;
scanf("%d %d", &N, &M);
Create(); /*堆初始化*/
int i;
int item;
for(i = 0; i < N; i++) { /*以逐个插入方式建堆*/
scanf("%d", &item);
Insert(item);
}
int j;
for(i = 0; i < M; i++) {
scanf("%d", &j);
printf("%d", H[j]);
while(j > 1) { /*沿根方向输出各结点*/
j /= 2;
printf(" %d", H[j]);
}
printf("\n");
}
return 0;
}
运行结果: