算法导论 最优二叉搜索树

#include <stdio.h>
#include <stdlib.h>
#include <float.h>

float ** optimalBST(float p[],float q[],int n,int **root)
{
	float **e=(float **)malloc((n+1)*sizeof(float*));
	float **w=(float **)malloc((n+1)*sizeof(float*));
	for(int i=0;i<n+1;i++)
	{
		e[i]=(float*)malloc((n+1)*sizeof(float));
		w[i]=(float*)malloc((n+1)*sizeof(float));
	}
	for(int i=1;i<n+1;i++)
	{
		e[i][i-1]=q[i-1];
		w[i][i-1]=q[i-1];
	}
	for(int l=1;l<n;l++)
	{
		for(int i=1;i<n-l+1;i++)
		{
			int j=i+l-1;
			e[i][j]=FLT_MAX;
			w[i][j]=w[i][j-1]+p[j]+q[j];
			for(int r=i;r<=j;r++)
			{
				float t=e[i][r-1]+e[r+1][j]+w[i][j];
				if(t<e[i][j])
				{
					e[i][j]=t;
					root[i][j]=r;
				}
			}
		}
	}
	return e;
}

void printE(float **e,int n)
{
	for(int i=1;i<n+1;i++)
	{
		for(int j=i-1;j<n;j++)
		{
			if(j==i-1)
				for(int k=0;k<i-1;k++)
					printf("     ");
			printf("%.2f ",e[i][j]);
		}
		printf("\n");
	}
}

void printRoot(int **root,int n)
{
	for(int i=1;i<n;i++)
	{
		for(int j=i;j<n;j++)
		{
			if(j==i)
				for(int k=0;k<i;k++)
					printf("  ");
			printf("%d ",root[i][j]);
		}
		printf("\n");
	}
}

void constructOptimalBST(int **root,int i,int j,int pi,int pj,char c,int n)
{
	if(i>j)
	{
		if(c=='L')
			printf("d%d为k%d的左孩子\n",j,root[i][i]);
		else
			printf("d%d为k%d的右孩子\n",j,root[j][j]);
		return;
	}

	if(c=='M')
		printf("k%d为根\n",root[i][j]);
	else 
		{
			
			if(c=='L')
				printf("k%d为k%d的左孩子\n",root[i][j],root[pi][pj]);
			else
				printf("k%d为k%d的右孩子\n",root[i][j],root[pi][pj]);
			
	}
	constructOptimalBST(root,i,root[i][j]-1,i,j,'L',n);
	constructOptimalBST(root,root[i][j]+1,j,i,j,'R',n);
}

void main()
{
	int n=6;
	int **root=(int **)malloc(n*sizeof(int*));
	for(int i=0;i<n;i++)
	{
		root[i]=(int*)malloc(n*sizeof(int));
	}
	float p[6]={0.0f,0.15f,0.1f,0.05f,0.1f,0.2f};
	float q[6]={0.05f,0.1f,0.05f,0.05f,0.05f,0.1f};
	float **e=optimalBST(p,q,n,root);
	printE(e,n);
	printRoot(root,n);
	constructOptimalBST(root,1,n-1,1,n-1,'M',n-1);
	getchar();
}