7-5 堆中的路径(最小堆)

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

输入格式:

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

输出格式:

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

 
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <stack>
#include <cmath>
using namespace std;
vector<int> v;
void upAdjust(int i){
    if(i==1) return;
    while(i!=1){
        if(v[i]<v[i/2]){
            swap(v[i],v[i/2]);
            i/=2;
        }else{
            break;
        }
    }
}
int main()
{
    int n,m;
    cin>>n>>m;
    v.resize(n+1);
    for(int i=1;i<=n;i++){
        cin>>v[i];
        upAdjust(i);
    }
    for(int i=0;i<m;i++){
        int num;
        cin>>num;
        int flag=1;
        while(num!=0){
            if(flag) {
                cout<<v[num];
                flag=0;
            }else {
                cout<<" "<<v[num];
            }
            num/=2;
        }
        cout<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_42068614/article/details/85558803
7-5