56 lines
1.4 KiB
Java
56 lines
1.4 KiB
Java
|
//给你两个 稀疏矩阵 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 |
|
|||
|
|
//
|
|||
|
|
// Related Topics 数组 哈希表 矩阵 👍 55 👎 0
|
|||
|
|
|
|||
|
|
package leetcode.editor.cn;
|
|||
|
|
|
|||
|
|
//311:稀疏矩阵的乘法
|
|||
|
|
class SparseMatrixMultiplication {
|
|||
|
|
public static void main(String[] args) {
|
|||
|
|
//测试代码
|
|||
|
|
Solution solution = new SparseMatrixMultiplication().new Solution();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//力扣代码
|
|||
|
|
//leetcode submit region begin(Prohibit modification and deletion)
|
|||
|
|
class Solution {
|
|||
|
|
public int[][] multiply(int[][] mat1, int[][] mat2) {
|
|||
|
|
int[][] result = new int[mat1.length][mat2[0].length];
|
|||
|
|
for (int i = 0; i < result.length; i++) {
|
|||
|
|
for (int j = 0; j < result[0].length; j++) {
|
|||
|
|
for (int k = 0; k < mat1[0].length; k++) {
|
|||
|
|
result[i][j] += mat1[i][k] * mat2[k][j];
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
//leetcode submit region end(Prohibit modification and deletion)
|
|||
|
|
|
|||
|
|
}
|