手机无码在线,天天碰天天摸,免费黄色成人视频网址,激情尻逼久久,黄色电影A片电影,亚洲综合五月婷婷,亚洲第一夜综合,黄色视频在线免费观看视频

算法的重要性，我就不多說了吧，想去大廠，就必須要經(jīng)過基礎(chǔ)知識和業(yè)務(wù)邏輯面試+算法面試。所以，為了提高大家的算法能力，這個公眾號后續(xù)每天帶大家做一道算法題，題目就從LeetCode上面選！

今天和大家聊的問題叫做 稀疏矩陣的乘法，我們先來看題面：

https://leetcode-cn.com/problems/sparse-matrix-multiplication/

Given two sparse matrices A and B, return the result of AB.
You may assume that A's column number is equal to B's row number.

給你兩個稀疏矩陣 A 和 B，請你返回 AB 的結(jié)果。

你可以默認 A 的列數(shù)等于 B 的行數(shù)。

示例

解題

由于是稀疏數(shù)組可以只計算非零數(shù)字。

將A矩陣轉(zhuǎn)存為hashmap,然后對每一個非零數(shù)遍歷B的列數(shù)來更新它在ans矩陣里可以被用到的位置。

class Solution {
public:
    vector<vector<int>> multiply(vector<vector<int>>& A, vector<vector<int>>& B) {
        int n = A.size(), m = A[0].size(), k = B[0].size();
        vector<vector<int>> res(n, vector<int>(k, 0));

        unordered_map<int, int> am;

        for (int i = 0; i < n; i ++)
            for (int j = 0; j < m; j ++)
            {
                if (!A[i][j]) continue;
                int idx = i * m + j;
                am[idx] = A[i][j];
            }

        for (auto aij : am)
        {
            int va = aij.second, aidx = aij.first, ai = aidx / m, aj = aidx % m;
            for (int j = 0; j < k; j ++)
            {
                res[ai][j] += va * B[aj][j];
            }
        }

        return res;
    }
};