05-树7 堆中的路径

题目

将一系列给定数字插入一个初始为空的小顶堆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;
}

运行结果：