7-5 堆中的路徑(最小堆)
阿新 • • 發佈:2019-01-05
將一系列給定數字插入一個初始為空的小頂堆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; }