05-树7 堆中的路径

最新推荐文章于 2024-07-23 00:37:07 发布

原创最新推荐文章于 2024-07-23 00:37:07 发布 · 129 阅读

0 ·

CC 4.0 BY-SA版权

PTA 专栏收录该内容

17 篇文章

订阅专栏

该程序实现了一个小顶堆的构建过程，通过插入一系列整数将空堆填充。然后，对于给定的下标，输出从该下标到堆根节点的路径。程序包括创建空堆、插入元素和打印路径的函数。测试用例中包含N个插入元素和M个需要打印的路径下标。

将一系列给定数字插入一个初始为空的小顶堆H[]。随后对任意给定的下标i，打印从H[i]到根结点的路径。

输入格式:

每组测试第1行包含2个正整数N和M(≤1000)，分别是插入元素的个数、以及需要打印的路径条数。下一行给出区间[-10000, 10000]内的N个要被插入一个初始为空的小顶堆的整数。最后一行给出M个下标。

输出格式:

对输入中给出的每个下标i，在一行中输出从H[i]到根结点的路径上的数据。数字间以1个空格分隔，行末不得有多余空格。

#include <iostream>
using namespace std;

#define Max 1001
#define MinV -10001
class Heap{
public:
    int heap[Max];   //从1开始储存，heap[0]作为哨兵
    int size;        //堆中元素个数
};
//建空堆，初始化
Heap createHeap();
//向堆中插入数据
void insertHeap(Heap& h, int data);
//打印路径
void printRode(Heap h, int index);

int main()
{
    Heap h = createHeap();
    int N, M;   // N个数  M为需打印的路径条数
    cin >> N >> M;
    for(int i = 0; i < N; i++){
        int data;
        cin >> data;
        insertHeap(h, data);
    }
    //打印路径
    for(int i = 0; i < M; i++){
        int index;
        cin >> index;
        printRode(h, index);
        if(i != M - 1) cout << endl;
    }
    
    return 0;
}

//建空堆，初始化
Heap createHeap(){
    Heap h;
    h.heap[0] = MinV;
    h.size = 0;
    return h;
}

//向堆中插入数据
void insertHeap(Heap& h, int data)
{
    if(h.size == Max - 1) return;    //堆满
    int i;
    for(i = ++h.size; h.heap[i/2] > data; i /= 2){
        h.heap[i] = h.heap[i/2];
    }
    h.heap[i] = data;
}

//打印路径
void printRode(Heap h, int index){
    if(index < 1) return;
    cout << h.heap[index];
    for(int i = index/2; i >= 1; i /= 2){
        cout << " " << h.heap[i];
    }
}