最优矩阵链乘
[题目]
一个n*m的矩阵由n行m列共n*m排列而成。两个矩阵A和B可以相乘当且仅当A的列数等于B的行数。一个n*m的矩阵乘m*p的矩阵,运算量为n*m*p。
矩阵乘法不满足分配律,但满足结合律。因此A*B*C既可以按顺序(A*B)C也可以按A(B*C)来进行。假设A、B、C分别是2*3、3*4、4*5的,则(A*B)C运算量是2*3*4+2*4*5=64,A(B*C)的运算量是3*4*5*2*3*5=90.显然第一种顺序节省运算量。
给出n个矩阵组成的序列,设计一种方法把他们依次乘起来,使得总的运算量尽量小。假设第i个矩阵A[i]是P[i-1]*P[i]的。
输入
3
2 3
3 4
4 5
输出
64
[分析]
矩阵链乘可以看成一个表达式,无论加多少括号,一定存在最后一次乘法,而最后一次乘号的左右两边的子表达式一定存在最优解(运算量最小),也就是最优子结构。状态转移方程如下:
f(i,j)=min{f(i,k)+f(k+1,j)+pi-1pkpj};
其中f(i,j)表示第i个矩阵到第j个矩阵的运算量,k为中间乘号的位置,p保存矩阵的行和列,第i个矩阵的行为pi-1,列为pi(这里用到了一个小技巧,由于输入是每一个矩阵的行和列,且前一个矩阵的列和后一个的行相同,在保存时重合读入)。
由于算法复杂度为O(n3),采用记忆化搜索优化。
[代码]