题目描述
使用动态规划算法求解矩阵连乘问题。
输入
每组数据包括两行,第一行为数组长度n,第二行为存储矩阵维数的一维数组。
输出
矩阵连乘最优计算次序。
样例输入 Copy
7
30 35 15 5 10 20 25
样例输出 Copy
A[2:2] * A[3:3]
A[1:1] * A[2:3]
A[4:4] * A[5:5]
A[4:5] * A[6:6]
A[1:3] * A[4:6]
*求最优解就是把加括号过程写出来*
import java.util.Scanner;
public class juzhenliancheng3 {
static int m[][];
static int p[];
static int n;
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner sc=new Scanner(System.in);
while(sc.hasNext()) {
n=sc.nextInt();
p=new int[n];
m=new int[n][n];
int s[][]=new int[n][n];
/*
* for(int i=0;i<n;i++) { for(int j=0;j<n;j++) { m[i][j]=0; } }
*/
for(int i=0;i<n;i++) {
p[i]=sc.nextInt();
}
matrixChain3(n,p,m,s);
traceback(s,1,n-1);
}
}
//在之前求最优值的基础上用s数组保存求m[i][j]时最优的分割位置
private static void matrixChain3(int n,int[] p, int[][] m,int[][] s) {
// TODO Auto-generated method stub
int n1=n-1;
for(int i=1;i<=n1;i++) {m[i][i]=0;}//主对角线置0,m数组从(1,1)开始放元素
for(int r=2;r<=n1;r++) {//第r行对角线
for(int i=1;i<=n1-r+1;i++) {//每条对角线从第0行到第n-r行
int j=r-1+i;
m[i][j]=m[i][i]+m[i+1][j]+p[i-1]*p[i]*p[j];//初始化每个m[i][j]看作最小
s[i][j]=i;
for(int k=i+1;k<j;k++) {
int f=m[i][k]+m[k+1][j]+p[i-1]*p[k]*p[j];
if(f<m[i][j]) {//找到更小的赋给m[i][j]
m[i][j]=f;
s[i][j]=k;
}
}
}
}
}
//用递归来实现二叉树的自底向上的过程
private static void traceback(int[][] s, int i, int j) {
// TODO Auto-generated method stub
if(i==j) return;//一直二叉树分下去,结束条件就是分到一个矩阵时
traceback(s,i,s[i][j]);
traceback(s,s[i][j]+1,j);
int q=s[i][j]+1;
System.out.println("A["+i+":"+s[i][j]+"] * A["+q+":"+j+"]");
}
}