所谓矩阵链乘法是指当一些矩阵相乘时,如何加括号来改变乘法顺序从而来降低乘法次数。例如有三个矩阵连乘:A1⋅A2⋅A3,其维数分别为:10×100,100×5,5×50. 如果按照((A1⋅A2)⋅A3)来计算的话,求(A1⋅A2)要10⋅100⋅5=5000次乘法,再乘以A3需要10⋅5⋅50=2500次乘法,因此总共需要7500次乘法。如果按照(A1⋅(A2⋅A3))来计算的话,求(A2⋅A3)要100⋅5⋅50=25000次乘法,再乘以A1需要10⋅100⋅50=50000次乘法,因此总共需要75000次乘法。可见,按不同的顺序计算,代价相差很大。
矩阵链乘法问题可以表述如下:给定n个矩阵构成的一个链(A1⋅A2⋅A3⋅…⋅An),其中i=1,2,……n,矩阵Ai的维数为p(i−1)⋅p(i),对于乘积A1⋅A2⋅A3⋅…⋅An以一种最小化标量乘法次数的方式进行加括号。
解决这个问题,我们可以用穷举法,但是n很大时,这不是个好方法,其时间复杂度为指数形式。拿上面的例子来说,加括号后把矩阵链分成了两部分,计算代价为两者代价的和。因此假设这种方法的代价最少,则两个部分的代价也是最小的,如果不是最小的,那么这种方法就不是最优的,因此矩阵链乘法具有最优子结构。因此我们可以利用子问题的最优解来构造原问题的一个最优解。所以,可以把问题分割为两个子问题$(A1\cdot A_2\cdot A_3\cdot\ldots\cdot A_k和A{k+1}\cdot A{k+2}\cdot A{k+3}\cdot\ldots\cdot A_n),需找子问题的最优解,然后合并这些问题的最优解。从下面的程序可以看出,其时间复杂度为O(n^3)$.
上面算法的实现程序如下:
1 |
|
在上面程序的实现中,矩阵链的参数如下表:
matrix | dimension |
---|---|
A1 | 30 × 35 |
A2 | 35 × 15 |
A3 | 15 × 5 |
A4 | 5 × 10 |
A5 | 10 × 20 |
A6 | 20 × 25 |
表示第i个矩阵到第j个矩阵的计算代价矩阵m[i][j]和表示第i个矩阵到第j个矩阵的最小代价时的分为两部分的位置矩阵s[i][j]的结果如下图:
从上面左图的m矩阵可以看出任意第i个到第j个矩阵连乘的乘法次数。最终的加括号形式为:(A1(A2A3))((A4A5)A6)
用动态规划算法解矩阵链乘法问题需要时间为O(n3),空间为O(n2),这比采用穷举法的指数时间相比要有效的多。