求问， sparse dot vector multiplication

经常看到FB的一道题， sparse dot vector multiplication

我觉得应该两种cases:
case1:  (1*n)  * (n*1) = (1)  
case2:  (n*1) * (1*n) = (n*n),  //case2 其实就是 sparse matrix multiply.

所以应该是 Sparse dot vector multiplication 指的应该是case1； 但是感觉代码应该很简单的，

1.         public int vectorMultiply(int[] A, int B[]){
   
2.                 int n = A.length;
   
3.                 int result = 0;
   
4.                
   
5.                 for(int i=0; i<n; i++){
   
6.                         if(A[i] == 0 || B[i]==0)
   
7.                                 continue;
   
8.                        
   
9.                         result += A[i] * B[i];
   
10.                 }
    
11.                 return result;
    
12.         }

复制代码

请问我的代码有问题吗， 这个问题有什么tricky的地方吗？
谢谢

这个问题主要的考察点应该在于如何表示sparse vector/matrix，以及如何用sparse representation来进行乘法。如果已经给出dense representation，其实没什么太大意义。

同意楼上的，这么做没有利用上sparse的条件

这个题目，
你的输入应该是两个已经用 （index ，value）存好的sparse list

你应该把输入改成两个2d array，里面随机填写0 和 1，然后输出是另外一个2d array