矩阵向量乘法CCS c++

Question

我试图用压缩柱存储中的矩阵来乘矩阵和向量。汇总表是：

0   3   0   
4   0   0   
2   0   0

以下是CCS表格：

[ 4 2 3 ]
[ 2 3 1 ]
[ 1 3 4 4 ]

矢量是：

[ 1 3 4 ]

所以产品应该是

[ 9 4 2 ]

下面是我要创建的函数

vector<int> multiply(vector<int> val,vector<int> col,vector<int> row,vector<int> v, int r){

    vector<int> x(v.size());
    for (int j=0; j<r; j++) {
        for (int i=col[j]-1; i<col[j+1]-1; i++) {
            cout<<"\n"<<val[i]<<"*"<<v[j]<<"\n";
            x[j]+= val[i]*v[j];
        }
    }
    return x;
}

但这又回来了

[ 6 9 0 ]

这是我得到的最接近真正的解决方案，我怎样才能解决这个问题？

Answer 1

我认为这是由col_ptr矢量驱动的。

1.)我不确定r可以取多少值，所以我删除了它，因为我们不需要这些信息来解决问题2)。我应该注意，我没有编译这个，但是我相信算法是正确的。)有一些明显的方法可以优化这段代码的内存使用，但是我保留了这些变量来帮助解释这个过程。4.)您发布的代码的主要问题是，它似乎没有使用行数组来告诉我们值在哪一行！

vector<int> multiply(vector<int> val,vector<int> col,vector<int> row,vector<int> v) {
   vector<int> x(v.size());
   //we may need to initialize x as all zeroes, I forget
   int col_size = col.size();
   int column_idx = 0;
   for (int j=0; j<col_size-1; j++) {
      //j indicates where the columns start going!
      //columns end prior to the next column start
      //val_index is where we need to start in the val array, for this column
      int val_index_start = col[j]-1; //this is redunda
      //we keep traversing forward until the next column starts
      int next_column_start = col[j+1]-1;
      for (int k=val_index_start; k < next_column_start && k < v.size(); k++) {
        int row_of_cell = row[k] - 1;
        int column_of_cell = column_idx;
        int product = v[column_idx]*val[k];
        x[row_of_cell] += product;
      }
      column_idx++;
   }
   return x;
}