Sparse Matrix Multiplication

原创 2016年08月30日 08:05:27

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.

Example:

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 |
思路1:矩阵相乘就是所有的k,A(i,k) * B(k,j) = C(i,j) ,稀疏矩阵就是 有很多0,为了提高速度也就是如果 A(i,k) 或者B(k,j), k 从0到length,如果有0,那么这个计算就不进行了。(那么我们来看结果矩阵中的某个元素C[i][j]是怎么来的,起始是A[i][0]*B[0][j] + A[i][1]*B[1][j] + ... + A[i][k]*B[k][j],那么为了不重复计算0乘0,我们首先遍历A数组,要确保A[i][k]不为0,才继续计算,然后我们遍历B矩阵的第k行,如果B[K][J]不为0,我们累加结果矩阵res[i][j] += A[i][k] * B[k][j]; 这样我们就能高效的算出稀疏矩阵的乘法)

public class Solution {
    public int[][] multiply(int[][] A, int[][] B) {
        int[][] C = new int[A.length][B[0].length];
        for(int i=0; i<A.length; i++){
            for(int k=0; k<A[0].length; k++){
                if(A[i][k] != 0) {
                    for(int j=0; j<B[0].length; j++){
                        if(B[k][j] != 0){
                            C[i][j] += A[i][k] * B[k][j];
                        }
                    }
                }
            }
        }
        return C;
    }
}
思路2:就是稀疏矩阵的表达方式就是记录所有不为0的元素的位置,然后再进行计算。我刚开始想到用hashmap,但是iterator不好用,其实后来看了别人的答案,觉得自己笨了,其实list就够了,只要判断前后相等的元素进行运算就行了。按道理来说,这个应该更快,如果矩阵很大的话;但是这个是150ms的运算速度。上面那个是60ms。

public class Solution {
    
    class Node{
        int x;
        int y;
        public Node(int x, int y){
            this.x = x;
            this.y = y;
        }
    }
    
    public int[][] multiply(int[][] A, int[][] B) {
        int[][] C = new int[A.length][B[0].length];
        List<Node> listA = new ArrayList<Node>();
        List<Node> listB = new ArrayList<Node>();
        for(int i=0; i<A.length; i++){
            for(int j=0; j<A[0].length; j++){
                if(A[i][j]!=0){
                    listA.add(new Node(i,j));
                }
            }
        }
        
        for(int i=0; i<B.length; i++){
            for(int j=0; j<B[0].length; j++){
                if(B[i][j]!=0){
                    listB.add(new Node(i,j));
                }
            }
        }
        
        for(Node a: listA){
            for(Node b: listB){
                if(a.y == b.x){
                    C[a.x][b.y] += A[a.x][a.y]*B[b.x][b.y];
                }
            }
        }
        return C;
    }
}





版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

[Leetcode] 311. Sparse Matrix Multiplication 解题报告

题目: Given two sparse matrices A and B, return the result of AB. You may assume that A's co...

LeetCode 311. Sparse Matrix Multiplication(稀疏矩阵相乘)

原题网址:https://leetcode.com/problems/sparse-matrix-multiplication/ Given two sparse matrices A and B...
  • jmspan
  • jmspan
  • 2016年04月21日 08:12
  • 1344

[leetcode] 311. Sparse Matrix Multiplication 解题报告

题目链接: https://leetcode.com/problems/sparse-matrix-multiplication/ Given two sparse matrices A and ...

1st round, 311 Sparse Matrix Multiplication

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

分布式稀疏矩阵的数据结构(data structure for distributed sparse matrix)

研究下sparse matrix, 最近正好看了一个项目: hpcg。 一周多时间,感觉蛮有意思的。rebuild the wheel....

稀疏矩阵的存储格式(Sparse Matrix Storage Formats)

对于很多元素为零的稀疏矩阵,仅存储非零元素可使矩阵操作效率更高。现有许多种稀疏矩阵的存储方式,但是多数采用相同的基本技术,即存储矩阵所有的非零元素到一个线性数组中,并提供辅助数组来描述原数组中非零元素...

稀疏矩阵的存储格式(Sparse Matrix Storage Formats)

稀疏矩阵的存储格式(Sparse Matrix Storage Formats) Sason@CSDN   对于很多元素为零的稀疏矩阵,仅存储非零元素可使矩阵操作效率更高。现有许多种稀疏矩阵的存...

稀疏矩阵的存储格式(Sparse Matrix Formats)

稀疏矩阵的存储格式(Sparse Matrix Storage Formats)1.Coordinate Format(COO) 这种存储方式的主要优点是灵活、简单。仅存储非零元素以及每个非零元素的...

稀疏矩阵(Sparse Matrix)

稀疏矩阵(Sparse Matrix) 注:压缩存储的矩阵可以分为特殊矩阵和稀疏矩阵。对于那些具有相同元素或零元素在矩阵中分布具有一定规律的矩阵,被称之为特殊矩阵。对于那些零元素数据远远多于非零元素...
  • cainv89
  • cainv89
  • 2016年06月29日 21:12
  • 2400

matlab-sparse函数和full函数-sparse matrix和full matrix

sparse函数 功能:Create sparse matrix-创建稀疏矩阵 用法1:S=sparse(X)——将矩阵X转化为稀疏矩阵的形式,即矩阵X中任何零元素去除,非零元素及其下标(索引)组...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:Sparse Matrix Multiplication
举报原因:
原因补充:

(最多只允许输入30个字)