算法导论 练习题 15.5-4

#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];
			//初始化root边际情况，用于递推的开始。
			//只有一个结点，root[i][j]就是自身
			if(i==j)
			{
				e[i][j]=e[i][i-1]+e[i+1][j]+w[i][j];
				root[i][j]=i;
			}
			else
			{
				//根据题目里给的条件，root[i][j]一定在root[i][j-1]和root[i+1][j]之间
				//所以可以缩小r的取值范围
				for(int r=root[i][j-1];r<=root[i+1][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();  
}