算法导论 最优二叉查找树

#include<iostream>
using namespace std;
const int N=100;
double p[N]={0};
double q[N]={0};
double e[N][N]={0.0};//e[i][j]为搜索一棵包含关键字ki...kj的最有二叉查找树的期望代价
double w[N][N]={0.0};//w[i][j]为一棵包含关键字ki...kj的子树的概率总和
int root[N][N]={0};//root[i][j]为kr的下标r
void OPTIMAL_BFS(double p[],double q[],int n)
{
	int i,j,r,l;//l为树的关键字的个数
	double t;
	for(i=1;i<=n+1;i++)
	{
		e[i][i-1]=q[i-1];
		w[i][i-1]=q[i-1];
	}
	for(l=1;l<=n;l++)
		for(i=1;i<=n-l+1;i++)
		{
			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;r++)
			{
				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;
				}
			}
		}
}

int main()
{
	int n,i,j;//n为关键字的个数
	cout<<"Enter n: ";
	cin>>n;
	for(i=1;i<=n;i++)
		cin>>p[i];
	for(i=0;i<=n;i++)
		cin>>q[i];
	OPTIMAL_BFS(p,q,n);
	for(i=1;i<=n+1;i++)
		for(j=n;j>=i-1;j--)
		{
			cout<<e[i][j]<<' ';
			if(j%5==0)
				cout<<endl;
		}
	cout<<endl;
	for(i=1;i<=n+1;i++)
		for(j=n;j>=i-1;j--)
		{
			cout<<w[i][j]<<' ';
			if(j%5==0)
				cout<<endl;
		}
	for(i=1;i<=n;i++)
		for(j=n;j>=i;j--)
		{
			if(j%5==0)
				cout<<endl;
			cout<<root[i][j]<<' ';
		}
	return 0;
}