leetcode Sparse Matrix Multiplication 稀疏矩阵相乘

最新推荐文章于 2021-10-19 23:28:49 发布

acttell

最新推荐文章于 2021-10-19 23:28:49 发布

阅读量511

点赞数

分类专栏：数组

本文链接：https://blog.youkuaiyun.com/acttell/article/details/80810108

版权

数组专栏收录该内容

49 篇文章

订阅专栏

本文介绍了一种矩阵乘法的优化方法，通过避免与零元素相乘来减少不必要的计算。该方法首先遍历第一个矩阵A，仅当A[i][k]不为0时进行下一步计算；随后遍历第二个矩阵B，若B[k][j]不为0，则更新结果矩阵C的对应元素C[i][j] += A[i][k] * B[k][j]。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

     |  1 0 0 |   | 7 0 0 |   |  7 0 0 |
AB = | -1 0 3 | x | 0 0 0 | = | -7 0 3 |
                  | 0 0 1 |
思路：i x k维度矩阵A与 k x j维度矩阵B 相乘得i x j维度矩阵C，
C[i][j]=A[i][0]*B[0][j] +... + A[i][k]*B[k][j]，
为了不重复含0相乘无用功，先遍历A，A[i][k]不为0才计算，
再遍历B,若B[k][j]不为0，c[i][j] += A[i][k] * B[k][j]