5-5 堆中的路径 (25分) -- 小堆树性质，如何调整

5-5 堆中的路径   (25分)

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

输入格式:

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

输出格式:

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

输入样例:

5 346 23 26 24 105 4 3

输出样例:

24 23 1046 23 1026 10

 

• 时间限制：400ms
• 内存限制：64MB
• 代码长度限制：16kB
• 判题程序：系统默认
• 作者：陈越
• 单位：浙江大学

题目判定

/* 5-5 堆中的路径   (25分)http://pta.patest.cn/pta/test/15/exam/4/question/713*/  #include <cstdio>  #include <cstdlib>  #include <iostream>  #include <vector>  #include <queue>  #include <map>#include <stack>#include <iterator>#include <algorithm>    using namespace std;  #define N 1005int a[N] ;void adjust(int k){while(k != 1){int fa = k / 2 ;if ( a[k] < a[fa]){int tmp = a[k] ;a[k] = a[fa] ;a[fa] =  tmp ;}k = fa ;}}int main()  {      //freopen("in.txt","r",stdin);      //freopen("out.txt","w",stdout);  int i , n , m , tmp;scanf("%d%d",&n,&m);for(i = 1;i <= n;i++){scanf("%d",&tmp);a[i] = tmp ;adjust(i);}vector<int> v ;for(i = 0 ;i < m ; i++){scanf("%d",&tmp);v.clear();int len = 0;int k = tmp ;while(k != 0){v.push_back(a[k]);len ++ ;k = k / 2 ;}printf("%d",v[0]);for(int j = 1 ; j < len ; j++){printf(" %d",v[j]);}printf("\n");}    return 0 ;  }