假设有六个矩阵连乘,如下
A1 A2 A3 A4 A5 A6
30*35 35*15 15*5 5*10 10*20 20*25
想要求得这六个矩阵相乘的最优解,可以使用动态规划的方法解决。动态规划是指将待求解问题分解成若干个子问题,先求得子问题,然后从这些子问题的解得到原问题的解。
步骤如下:
①分析最优解结构,建立递归关系 ②计算最优值 ③构造最优解
一、构建数组分别记录矩阵的行列数,连乘次数和最佳分割位置
//记录数组行列数量
int p[]={30,35,15,5,10,20,25};
//记录连乘次数
int b[][]=new int[p.length][p.length];
//记录最佳分割位置
int s[][]=new int[p.length][p.length];
二、初始化对角线为0,从第二列开始算出对角线的值,其中连乘次数m[i][j]=m[i][k]+m[k+1][j]+p(i-1)*p(k)*p(j)
public static void matrixChain(int[] p,int[][] m,int[][] s){
int n=p.length-1;
//初始化对角线的位置为0;
for(int i=1;i<=n;i++)
m[i][i]=0;
//从2列开始确定对角线以外的值
for(int r=2;r<=n;r++)
for(int i=1;i<=n-r+1;i++){
int j=i+r-1;
m[i][j]=m[i+1][j]+p[i-1]*p[i]*p[j];
s[i][j]=i;
for(int k=i+1;k<j;k++){
int t=m[i][k]+m[k+1][j]+p[i-1]*p[k]*p[j];
if(t<m[i][j]){
m[i][j]=t;
s[i][j]=k;
}
}
}
}
三、构造最优解
public static void traceback(int[][] s,int i,int j){
if(i==j){
System.out.print("A"+i);
return;
}
System.out.print("(");
traceback(s,i,s[i][j]);
traceback(s,s[i][j]+1,j);
System.out.print(")");
}
四、结果示例