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 解题报告

题目链接: https://leetcode.com/problems/sparse-matrix-multiplication/ Given two sparse matrices A and ...
  • qq508618087
  • qq508618087
  • 2016年03月13日 04:19
  • 2975

[leetcode 311] Sparse Matrix Multiplication

Question: Given two sparse matrices A and B, return the result of AB. You may assume that ...
  • Xiaohei00000
  • Xiaohei00000
  • 2015年12月29日 22:45
  • 1986

[LeetCode311]Sparse Matrix Multiplication

Given two sparse matrices A and B, return the result of AB.You may assume that A's column number is ...
  • u012175043
  • u012175043
  • 2015年11月29日 07:33
  • 1940

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
  • 1519

hdu 1082 Matrix Chain Multiplication

题目地址: http://acm.hdu.edu.cn/showproblem.php?pid=1082 题目描述: Matrix Chain Multiplication Time ...
  • hackerwin7
  • hackerwin7
  • 2014年01月27日 17:11
  • 1550

Matrix Chain Multiplication(栈的简单应用)

Matrix Chain Multiplication Time Limit: 2 Seconds      Memory Limit: 65536 KB Matrix multip...
  • cs_zlg
  • cs_zlg
  • 2013年01月12日 11:57
  • 458

Leetcode Sparse Matrix Multiplication

Given two sparse matrices A and B, return the result of AB. You may assume that A's column numb...
  • u011472124
  • u011472124
  • 2016年06月15日 11:50
  • 107

311. Sparse Matrix Multiplication

Given two sparse matrices A and B, return the result of AB. You may assume that A's column number ...
  • lilibaobao89
  • lilibaobao89
  • 2017年01月25日 08:12
  • 83

Sparse Matrix Multiplication

Problem Given two sparse matrices A and B, return the result of AB. You may assume that A'...
  • Scarlett_leetcode
  • Scarlett_leetcode
  • 2016年03月21日 09:12
  • 105

Leetcode: Sparse Matrix Multiplication

QuestionGiven two sparse matrices A and B, return the result of AB.You may assume that A’s column nu...
  • ayst123
  • ayst123
  • 2015年12月28日 11:02
  • 924
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:Sparse Matrix Multiplication
举报原因:
原因补充:

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