题目
将一系列给定数字插入一个初始为空的小顶堆H[]。随后对任意给定的下标i,打印从H[i]到根结点的路径。
输入格式:
每组测试第1行包含2个正整数N和M(≤1000),分别是插入元素的个数、以及需要打印的路径条数。下一行给出区间[-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
思路一:
将n个元素一个一个的插入到初始为空的堆中去,时间代价为O(l g n),在插入的过程就是先了小顶堆的实现方式。
代码一:
#include <iostream>
using namespace std;
const int maxdata=-10005;//最小堆的哨兵
struct Minheap{
int *data;
int size;
int capacity;
};
typedef Minheap* minheap;
void creatheap(minheap &heap,int maxsize){
heap=new(Minheap);
heap->data=(int *)malloc((maxsize+1)*sizeof(int));
heap->size=0;
heap->capacity=maxsize;
heap->data[0]=maxdata;//哨兵
}
bool isfull(minheap heap){
return heap->size==heap->capacity;
}
bool isempty(minheap heap){
return heap->size==0;
}
void insert(minheap &heap,int x){
if(isfull(heap))
cout<<"it is full"<<endl;
int i=++heap->size;
for(;heap->data[i/2]>x;i/=2)
heap->data[i]=heap->data[i/2];
heap->data[i]=x;
}
int main(){
int n,l,x;
cin>>n>>l;
minheap minh;
creatheap(minh,n);
while(n--){
cin>>x;
insert(minh,x);
}
for(int i=0;i<l;i++){
cin>>x;
cout<<minh->data[x];
for(int i=x/2;i>0;i/=2)
cout<<' '<<minh->data[i];
cout<<endl;
}
return 0;
}
思路二:但是在此题中答案是不对的!
因为从数据可以看出,是输入一个数排序一个数的。
如果需要用到堆,可以有更快的方式处理:
将n个元素先插入到二叉树当中去,要求就只是先满足二叉树的结构特征,不需要满足有序性。第二步就是调整各个结点的元素,满足最大堆的有序特征。从第N/2个结点(即最后一个有儿子的结点开始),对包括此结点以及它前面的各个结点,一直到根结点结束,逐一向下过滤操作。
#include <iostream>
using namespace std;
const int mindata=-10000;
struct Minheap{
int *Data;
int size;
int capacity;
};
typedef Minheap* minheap;
void creatminheap(minheap &heap,int maxsize){
heap=new(Minheap);
heap->Data=(int *)malloc((maxsize+1)*sizeof(int));
heap->size=0;
heap->capacity=maxsize;
heap->Data[0]=mindata;
}
void sortheap(minheap &heap,int i){
int child,parent;
int x=heap->Data[i];
for(parent=i;parent*2<=heap->size;parent=child){
child=parent*2;
if(child!=heap->size&&(heap->Data[child]>heap->Data[child+1]))
child++;
if(heap->Data[child]>=x)
break;
else
heap->Data[parent]=heap->Data[child];
}
heap->Data[parent]=x;
}
void bulidheap(minheap &heap){
for(int i=heap->size/2;i>0;i--)
sortheap(heap,i);
}
int main(){
int n,l,x;
cin>>n>>l;
minheap h;
creatminheap(h,n);
for(int i=1;i<=n;i++){
cin>>h->Data[i];
h->size++;
}
bulidheap(h);
for(int i=0;i<l;i++){
cin>>x;
cout<<h->Data[x];
for(int i=x/2;i>0;i/=2)
cout<<' '<<h->Data[i];
cout<<endl;
}
return 0;
}