矩阵相乘的strassen算法_矩阵乘法Strassen算法

阅读经典——《算法导论》03

矩阵乘法是种极其耗时的运算。

以C = A • B为例，其中A和B都是 n x n 的矩阵。根据矩阵乘法的定义，计算过程如下：

SQUARE-MATRIX-MULTIPLY(A, B)

n = A.rows

let C be a new nxn matrix

for i = 1 to n

for j = 1 to n

c[i][j] = 0

for k = 1 to n

c[i][j] += a[i][k] * b[k][j]

return C

由于存在三层循环，它的时间复杂度将达到O(n3)。

这是一个很可怕的数字。但是，凭着科学家们的智慧，这个数正在一步步下降。本文介绍经典的Strassen算法，该算法将时间复杂度降低到O(nlg7) ≈ O(n2.81)。别小看这个细微的改进，当n非常大时，该算法将比平凡算法节约大量时间。

分治法

Strassen算法基于分治的思想，因此我们首先考虑一个简单的分治策略。

每个 n x n 的矩阵都可以分割为四个 n/2 x n/2 的矩阵：

(式3-1)

因此可以将公式C = A • B改写为

(式3-2)

于是上式就等价于如下四个公式：

(式3-3)<