文章目錄

• • 1. 題目
  • 2. 解題
  • • 2.1 暴力求解
    • 2.2 選取都不為0的行和列相乘

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 |

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/sparse-matrix-multiplication
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

2.1 暴力求解

• 3層 for 循環

class Solution { public:vector<vector<int>> multiply(vector<vector<int>>& A, vector<vector<int>>& B) {vector<vector<int>> ans(A.size(), vector<int>(B[0].size(), 0));int i, j, k, sum;for(i = 0; i < A.size(); ++i)for(j = 0; j < B[0].size(); ++j){sum = 0;for(k = 0; k < A[0].size(); ++k)sum += A[i][k]*B[k][j];ans[i][j] = sum;}return ans;} };

24 ms 8.4 MB

2.2 選取都不為0的行和列相乘

class Solution { public:vector<vector<int>> multiply(vector<vector<int>>& A, vector<vector<int>>& B) {vector<bool> r_all0(A.size(),true);vector<bool> c_all0(B[0].size(),true);int i, j, k, sum;bool flag = false;for(i = 0; i < A.size(); ++i){flag = false;for(j = 0; j < A[0].size(); ++j){if(A[i][j]){flag = true;break;}}if(flag)r_all0[i] = false;}for(j = 0; j < B[0].size(); ++j){flag = false;for(i = 0; i < B.size(); ++i){if(B[i][j]){flag = true;break;}}if(flag)c_all0[j] = false;}vector<vector<int>> ans(A.size(), vector<int>(B[0].size(), 0));for(i = 0; i < A.size(); ++i)for(j = 0; j < B[0].size(); ++j){if(r_all0[i] || c_all0[j])continue;sum = 0;for(k = 0; k < A[0].size(); ++k)sum += A[i][k]*B[k][j];ans[i][j] = sum;}return ans;} };

8 ms 8.6 MB

---

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 311. 稀疏矩阵的乘法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。