Question

我想实现CSCmatrix（Compressed Sparse Coloumns）这样做我需要读取这种形式的矩阵：

{{1,2,3,4},{5,6,7,8}};

按列读取列！而不是逐行（在这种情况下，我可以使用像下面这样的简单的costructor）

template<typename T>
inline constexpr CSRmatrix<T>::CSRmatrix(std::initializer_list<std::initializer_list<T>>&& row ) noexcept
{
    this->rows = row.size();
    auto itr = *(row.begin());
    this->cols = itr.size();

    std::size_t i=0, j=0, w=0;  

    ia_.resize(rows+1);
    ia_[0] = 0;
    for(auto & r : row)
    {
        j=0 ; w =0 ;    
        for(auto & c : r)
        {
           if( c != 0.0 )
           {  
              a_.push_back(c);
             ja_.push_back(j);
             w++;
           }
           j++; 
        }
        i++;
        ia_[i] = ia_[i-1] + w ;
    }
}

逐行读取并以CSR格式存储矩阵（压缩稀疏行）可以帮助我吗？我没有想法！提前谢谢

我的数据将以{1,5,2,6,3,7,4,8}

的顺序存储在单个向量中

Answer 1

如果您的数据被声明为vector<vector<T>> data，则只需将data(v)置于初始化状态即可。但是，如果要切换行和列，请使用临时变量并切换行和列。

编辑或使用arr将数据放入单个数组

此代码假定初始化数据中的所有行大小相同

template<typename T> class matrix
{
public:
    //std::vector<std::vector<T>> data;
    std::vector<T> arr;
    matrix(std::initializer_list<std::vector<T>> v) //: data(v)
    {
        if(!v.size()) return; //invalid input
        std::vector<std::vector<T>> temp(v);
        //test to make sure all row are the same size
        //use the size of row[0] for all other rows:
        for(size_t c = 0; c < temp[0].size(); c++)
        {
            for(size_t r = 0; r < temp.size(); r++)
            {
                if(c < temp[r].size())
                    arr.push_back(temp[r][c]);
                //else {invalid input}
            }
        }
        for(size_t c = 0; c < temp[0].size(); c++)
            for(size_t r = 0; r < temp.size(); r++)
                arr.push_back(temp[r][c]);
    }
};

int main()
{
    matrix<int> m = { { 1,2,3,4 },{ 5,6,7,8 } };
    for(auto e : m.arr) std::cout << e << ",";
    std::cout << "\n";
    return 0;
}

输出

1,5,2,6,3,7,4,8，

c ++构造矩阵按列读取列

1 个答案: