LeetCode 311 稀疏矩阵的乘法

1. 题目

给你两个 稀疏矩阵 A 和 B，请你返回 AB 的结果。
你可以默认 A 的列数等于 B 的行数。

请仔细阅读下面的示例。

示例：
输入：
A = [
  [ 1, 0, 0],
  [-1, 0, 3]
]
B = [
  [ 7, 0, 0 ],
  [ 0, 0, 0 ],
  [ 0, 0, 1 ]
]
输出：
     |  1 0 0 |   | 7 0 0 |   |  7 0 0 |
AB = | -1 0 3 | x | 0 0 0 | = | -7 0 3 |
                  | 0 0 1 |

题解：

按照两个矩阵相乘的公式计算结果.

Since It is spase, skip k loop when A[i][j] == 0.

Time Complexity: O(m*n*o). m = A.length, n = A[0].length, o = B[0].length.

Space: O(1). regardless res.

class Solution {
    public int[][] multiply(int[][] A, int[][] B) {
        int m = A.length;
        int n = A[0].length;
        int o = B[0].length;

        int [][] res = new int[m][o];
        for(int i = 0; i<m; i++){
            for(int j = 0; j<n; j++){
                if(A[i][j] != 0){
                    for(int k = 0; k<o; k++){
                        res[i][k] += A[i][j]*B[j][k];
                    }
                }
            }
        }
        return res;
    }
}