[浙大数据结构] 05-树7 堆中的路径 (25分)

1 题目描述

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

2 解决思路

最小堆，根节点是最小的，父<子

新建一个堆栈

 //设置0，就是“岗哨”，防止越界
    H[0]=MINDATA;
    size =0;

然后写插入函数
从后往前写比较方便一点，然后比较父节点和当前值的大小，如果大于，则应该继续往前找，小于等于的话就应该插在这个位置上了。

3 解决代码

#include<stdio.h>
#define MAXN 1005
#define MINDATA -10005

int H[MAXN],size;
//插入数据建堆
void Insert(int x){
    int i =++size;
    //从后往前插,因为是最小堆，所以要求父<子
    //起点是最后一个元素的下一个，当前值x<父，往前挪，继续往前插
    for( ;x<H[i/2];i/=2){
        //父赋值给子结点
        H[i] = H[i/2];
    }
    //父亲小于等于x的时候，循环结束，当前父节点就是要插入的位置，直接赋值等于
    H[i] = x;
}

int main(){
    int i,N,M;
    int x;
    scanf("%d%d",&N,&M);
    //建空堆，堆初始化
    //设置0，就是“岗哨”，防止越界
    H[0]=MINDATA;
    size =0;
    //输入数据，插入数据建堆
    for(i =0;i<N;i++){
        scanf("%d",&x);
        Insert(x);
    }
    //M个查询
    for(i=0;i<M;i++){
        scanf("%d",&x);
        //当前节点的值先打印出来
        printf("%d",H[x]);
        //找父节点
        x/=2;
        while(x){
            printf(" %d",H[x]);
            x/=2;
        }
        printf("\n");
    }
    return 0;
}

4 提交结果