堆中的路径

堆中的路径

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

 

解题思路

　　主要是堆的插入操作。

　　AC代码：

#include <cstdio>

void insert(int *heap, int value, int pos);
void print(int *heap, int pos);

int main() {
    int n, m;
    scanf("%d %d", &n, &m);
    int heap[n + 1];
    for (int i = 1; i <= n; i++) {
        int value;
        scanf("%d", &value);
        insert(heap, value, i);
    }
    
    for (int i = 0; i < m; i++) {
        int pos;
        scanf("%d", &pos);
        print(heap, pos);
    }
    
    return 0;
}

void insert(int *heap, int value, int pos) {
    int i = pos;
    for ( ; i > 1 && value < heap[i >> 1]; i >>= 1) {
        heap[i] = heap[i >> 1];
    }
    heap[i] = value;
}

void print(int *heap, int pos) {
    for (int i = pos; i >= 1; i >>= 1) {
        printf("%d", heap[i]);
        if (i != 1) printf(" ");
        else printf("\n");
    }
}