算法导论 思考题 6-2 三叉堆实现

#include <stdio.h>
#include <limits.h>

#define D 3

int getMax(int a[],int k,int len)
{
	if(len<=0)
		return -1;
	int max=INT_MIN,p=0;
	for(int i=k;i<k+len;i++)
	{
		if(a[i]>max)
		{
			max=a[i];
			p=i;
		}
	}
	return p;
}

void MmaxHeapify(int a[],int k,int len)
{
	if(D*k-1>=len)
		return;
	int i,sublen,p;
	for(i=D*k-1,sublen=0;i<len && i<=D*k+1;i++,sublen++)
	{
		;
	}
	p=getMax(a,D*k-1,sublen);
	if(a[k]<a[p])
	{
		int temp=a[k];
		a[k]=a[p];
		a[p]=temp;
		MmaxHeapify(a,p,len);
	}
}

void MbuildHeap(int a[],int len)
{
	for(int i=len-1;i>=1;i--)
	{
		MmaxHeapify(a,i,len);
	}
}

void printA(int a[],int len)
{
	for(int i=1;i<len;i++)
	{
		printf("%d ",a[i]);
	}
	printf("\n");
}

int MextractMax(int a[],int len)
{
	int r=a[1];
	a[1]=a[len-1];
	MmaxHeapify(a,1,len-1);
	return r;
}

void MincreaseKey(int a[],int i,int key)
{
	if(key<a[i])
	{
		printf("新键必须大于原键！");
		return;
	}
	a[i]=key;		
	for(int j=(i+1)/D;j>=1;j=(i+1)/D)
	{
		if(a[j]>a[i])
			return;
		int temp=a[i];
		a[i]=a[j];
		a[j]=temp;
		i=j;
	}
}

void Minsert(int a[],int len,int key)
{
	a[len]=INT_MIN;
	MincreaseKey(a,len,key);
}


void main()
{
	int a[15]={INT_MAX,3,17,5,16,13,10,1,27,7,12,4,8,9,0};
	MbuildHeap(a,15);
	printA(a,15);
	MextractMax(a,15);
	printA(a,14);
	MincreaseKey(a,5,28);
	printA(a,14);
	Minsert(a,14,29);
	printA(a,15);
	getchar();
}