我正在尝试从随机数组构建最大堆,以便我可以在根中获得最大数量。
另外请告诉我,如何在heapify()函数中获取堆的大小。
#include<stdio.h>
#include<math.h>
int i;
void heapify(int a[],int i) //n=node at which heapify needs to apply
{
int size=8;
int max=i,l=2i,r=2i+1;
if(a[l]>a[max] && l<=size)
max=l;
if(a[r]>a[max] && r<=size)
max=r;
if(max==i) return;
else
a[max]^=a[i]^=a[max]^=a[i];
heapify(a,max);
}
//Using this function to call recursively heapify to maintain heap data structure
void build_heap(int arr[],int n) //n : size of array
{
for(i=floor(n/2);i>0;i--) // n/2 because after that every node is a leaf node and they follow heap property by default
{
heapify(arr,i);
}
}
int main()
{
int arr[] = {2,5,4,8,9,10,3,1};
build_heap(arr,8);
for(i=0;i<8;i++)
{
printf("%d ",arr[i]);
}
return 0;
}
答案 0 :(得分:0)
除了一些变化之外,您的代码似乎没有问题:
2i
和2i+1
更改为2*i
和2*i+1
,否则您将收到编译错误。实际上它应该是l = 2*i+1
和r = 2*i+2
,因为C数组是0-indexed
。您假设代码中有1-based
索引。build_heap()
中,运行[floor(n/2), 0]
的循环(您排除了0
&#39;元素可能是因为上述原因所述。)您可以将堆的大小作为参数传递给heapify()
函数,或将其声明为全局变量(尽管建议避免使用全局变量)。