力扣(LeetCode)算法_C++——稀疏矩阵的乘法

给定两个 稀疏矩阵 :大小为 m x k 的稀疏矩阵 mat1 和大小为 k x n 的稀疏矩阵 mat2 ,返回 mat1 x mat2 的结果。你可以假设乘法总是可能的。

示例 1:

输入:mat1 = [[1,0,0],[-1,0,3]], mat2 = [[7,0,0],[0,0,0],[0,0,1]]

输出:[[7,0,0],[-7,0,3]]

示例 2:

输入:mat1 = [[0]], mat2 = [[0]]

输出:[[0]]

cpp 复制代码
    vector<vector<int>> multiply(vector<vector<int>>& mat1, vector<vector<int>>& mat2) {
        int m = mat1.size();
        int k = mat1[0].size();
        int n = mat2[0].size();
        unordered_map<int, list<pair<int, int>>> map1;
        unordered_map<int, list<pair<int, int>>> map2;
        
        for (int i {0}; i < m; i++) {
            for (int j {0}; j < k; j++) {
                if (mat1[i][j] != 0) map1[mat1[i][j]].push_back(make_pair(i, j));
            }
        }        
        
        for (int i = 0; i < k; i++) {
            for (int j = 0; j < n; j++) {
                if (mat2[i][j] != 0) map2[mat2[i][j]].push_back(make_pair(i, j));
            }
        }

        vector<vector<int>> res(m, vector<int>(n, 0));
        if (m == 0 || n == 0) return res;

        for (auto& element1 : map1) {
            for (auto& list_node1 : element1.second) {
                for (auto& element2 : map2) {
                    for (auto& list_node2 : element2.second) {
                        if (list_node1.second == list_node2.first) {
                            res[list_node1.first][list_node2.second] += element1.first * element2.first;
                        }
                    }
                }
            }
            
        }

        return res;
    }
相关推荐
怀澈12214 分钟前
高性能服务器模型之Reactor(单线程版本)
linux·服务器·网络·c++
chnming198737 分钟前
STL关联式容器之set
开发语言·c++
带多刺的玫瑰41 分钟前
Leecode刷题C语言之统计不是特殊数字的数字数量
java·c语言·算法
爱敲代码的憨仔42 分钟前
《线性代数的本质》
线性代数·算法·决策树
威桑1 小时前
MinGW 与 MSVC 的区别与联系及相关特性分析
c++·mingw·msvc
熬夜学编程的小王1 小时前
【C++篇】深度解析 C++ List 容器:底层设计与实现揭秘
开发语言·数据结构·c++·stl·list
yigan_Eins1 小时前
【数论】莫比乌斯函数及其反演
c++·经验分享·算法
Mr.131 小时前
什么是 C++ 中的初始化列表?它的作用是什么?初始化列表和在构造函数体内赋值有什么区别?
开发语言·c++
阿史大杯茶1 小时前
AtCoder Beginner Contest 381(ABCDEF 题)视频讲解
数据结构·c++·算法
C++忠实粉丝1 小时前
计算机网络socket编程(3)_UDP网络编程实现简单聊天室
linux·网络·c++·网络协议·计算机网络·udp