1. 程式人生 > >7-5 堆中的路徑(最小堆)

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;
}