将一系列给定数字插入一个初始为空的小顶堆H[]
。随后对任意给定的下标i
,打印从H[i]
到根结点的路径。
输入格式:
每组测试第1行包含2个正整数N和M(≤),分别是插入元素的个数、以及需要打印的路径条数。下一行给出区间[-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
#include<iostream>
using namespace std;
#define maxn 1005
#define minh -10001
int H[maxn],siz;
int m;
int count=0;
void init()
{
siz=0;
H[0]=minh;//哨所
}
void Insert(int x)
{
siz++;
int i;
for(i=siz;H[i/2]>x;i/=2)
{
H[i]=H[i/2];
}
H[i]=x;
}
void Search(int x)
{
count++;
for(int i=x;i>0;i=i/2)
{
if(i==x)
cout<<H[x];
else
{
//i=i/2;
cout<<' '<<H[i];
}
//cout<<' '<<H[i/2];
}
if(count!=m)
cout<<endl;
}
int main()
{
int n,a,b;
string s;
cin>>n>>m;
init();
int i;
for( i=0;i<n;i++)
{
cin>>a;
Insert(a);
}
for(i=0;i<m;i++)
{
int t;
cin>>t;
Search(t);
}
return 0;
}
建立小根堆的代码
void init()//非常重要的一个函数,相当于一个哨所,两个元素,size 以及H[0]
{
siz=0;
H[0]=minh;//哨所
}
void Insert(int x)
{
siz++; //用szie来存储堆中的元素
int i;
for(i=siz;H[i/2]>x;i/=2) //循环每次都从要插入的位置开始,如果前面的数值比当前数值大,就把前面的数值赋给当前数值
{
H[i]=H[i/2];
}
H[i]=x; //当for循环中判断条件不满足时 ,将x赋给当前的i值
}
从当前路径寻找根路径的代码
void Search(int x)
{
count++;
for(int i=x;i>0;i=i/2)
{
if(i==x)
cout<<H[x];
else
{
//i=i/2;
cout<<' '<<H[i];
}
//cout<<' '<<H[i/2];
}
if(count!=m)
cout<<endl;
}