小白进阶之动态规划-最优二叉搜索树

问题描述：

给定关键词序列k=(k1,k2,...,kn)及其相应的概率pi，以及 n+1个伪关键字d0,d1,...,dn不在k值中。其中d0表示小于k1的值，dn表示大于kn的值，对于i=1,2,...,n-1，di表示在 ki和ki+1之间的值，用qi表示其搜索概率。用w(i,j)表示关键字ki,ki+1,...,kj的概率之和，e[i,j]表示包含关键字 ki,ki+1,...,kj的最优二叉搜索树中进行一次搜索的期望代价，root[j,i]记录包含关键字ki,...,kj的子树的根。

下面是源码：

optimalBst.h

#include<stdio.h>
#include<stdlib.h>
double optimalBST(double *p,double *q,int **root,int n);
void printBST(int **root,int i,int j);

optimalBst.cpp

#include"optimalBst.h"
double optimalBST(double *p,double *q,int **root,int n){
	double **e,**w;
	int i;
	e=(double **)malloc(sizeof(double *)*(n+2));
	w=(double **)malloc(sizeof(double *)*(n+2));
	for(i=0;i<n+2;i++){
		e[i]=(double *)malloc(sizeof(double)*(n+2));
		w[i]=(double *)malloc(sizeof(double)*(n+2));
	}
	for(i=0;i<n+2;i++){
		e[0][i]=0;
		e[i][n+1]=0;
		w[0][i]=0;
		w[i][n+1]=0;
	}
	for(i=1;i<n+2;i++){
		e[i][i-1]=q[i-1];
		w[i][i-1]=q[i-1];
	}
	int l;
	for(l=1;l<n+1;l++){
		for(i=1;i<=n-l+1;i++){
			int j,r;
			j=i+l-1;
			e[i][j]=INT_MAX;
			w[i][j]=w[i][j-1]+p[j]+q[j];
			for(r=i;r<j+1;r++){
				double 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[1][n];
}
void printBST(int **root,int i,int j){
	if(j==i-1)
		printf("d[%d]",j);
	else if(j==i){
		printf("d[%d]",i-1);
		printf("k[%d]",i);
		printf("d[%d]",i);
	}
	else{
		int r=root[i][j];
		printBST(root,i,r-1);
		printf("k[%d]",r);
		printBST(root,r+1,j);
	}
}

main.cpp

#include"optimalBst.h"
int main(){
	double p[6]={0,0.15,0.1,0.05,0.1,0.2};
	double q[6]={0.05,0.1,0.05,0.05,0.05,0.1};
	int **root;
	int n,i;
	n=5;
	root=(int **)malloc(sizeof(int *)*6);
	for(i=0;i<6;i++)
		root[i]=(int *)malloc(sizeof(int)*6);
	for(i=0;i<6;i++){
		root[0][i]=0;
		root[i][0]=0;
	}
	double res=optimalBST(p,q,root,n);
	printf("%lf\n",res);
	printBST(root,1,n);
	system("pause");
	return 0;
}