繁体   English   中英

C ++实现自定义矩阵类的迭代器

[英]c++ implementing iterators for custom matrix class

我目前正在研究用于教育目的和个人用途的线性代数库(自定义矢量和矩阵以及一些算法)。 我试图实现一个列迭代器,该遍历器遍历Matrix的指定列。

这是向量类的代码示例(基于该类的矩阵类):

template<class T>
class MVector
{
    std::vector<T> data;

public:

    explicit MVector(const std::size_t& n) :data(n) {}
    explicit MVector(std::size_t&& n) :data(n) {}

    typename std::vector<T>::iterator Begin(){
        return data.begin();
    }

    typename std::vector<T>::iterator End(){
        return data.end();
    }

    // many more functions and overloaded operators
    // end of class
};

矩阵类基于此向量(或与此有关的std :: vector),看起来像:

template<class T, std::size_t rowsize, std::size_t colsize>
class Matrix
{

private:

    // Data is stored in a MVector, a modified std::vector
    MVector<T> matrix;

    // size of row dimension of the matrix
    std::size_t row_dim;                  

    // size of row dimension of the matrix
    std::size_t column_dim;

public:

    Matrix(std::initializer_list<T> il) :matrix(il), 
                                     row_dim(rowsize), column_dim(colsize){}    
    //other constructors...

    // iterator
    typename std::vector<T>::iterator Begin(std::size_t row = 0){
        return matrix.Begin()+index(row,0);
    }

    typename std::vector<T>::iterator End(std::size_t row = rowsize){
        return matrix.Begin()+index(row,0);

    // index (convenience) function to access elements of the matrix via some_matrix(i,j)
    std::size_t index(std::size_t r, std::size_t c) const {
        return r*cols()+c; 
    }

    // this is exactly what I want the iterator to do: 
    // only without creating and returning an object. 

    // get c'th column
    // slicing is possible from both ends and by "jumping" over elements
    // @ param "begin" - starts at the n'th element 
    // @ param "end" - subtracts m from from the last element.
    // @ param "by" - selects every n'th column 
    MVector<T> get_column(std::size_t c, std::size_t begin = 0, 
                          std::size_t end = 0, std::size_t by = 1) const{
         assert(c < cols() && end < rows());
         MVector<T> columns;
         for (std::size_t i = index(begin, c); i < index(rows()-end,c); i+=by*cols()) {
             columns.addTo(matrix[i]);
         }
         return columns;                
    } 

// end of class  
};

因此,迭代行工作正常,我要做的就是:

 int main{

 Matrix<int, 3, 2> a = {1,2,3,4,5,6}; 
 for (std::vector<int>::iterator iter = a.Begin(1); iter != a.End(2); iter++) {
     std::cout << *iter << " ";
 }

 std::cout << endl;
 return 0;
 }

 bash-3.2$ ./main
 3 4 

这正是我想要的。 但是,遍历这些列不适用于该方法。 因此,我在寻找其他解决方案,并发现此Matrix类文章的Designing迭代器听起来与我的问题和情况非常相似,但是我无法得出该问题的解决方案。

其他建议指出了Boost库的迭代器:特别是:

boost::adaptors::stride(rng, n) 
boost::adaptors::slice(rng, n, m)

确实提供了非常相似的所需结果。 但是我的get_column函数也是如此。 但是,我不想创建一个新对象。 提升功能是做什么的。 来自文档“返回:基于rng的新范围,其中以n为步长执行遍历”。

因此,似乎迭代器不知道何时停止。

因此,我回到了第一个平方:如何返回迭代通过存储为向量的Matrix的列的迭代器?

我找到了解决问题的办法。 它是以下各项的组合:

this: 获取迭代器范围的每个第n个元素,以及this: http : //www.codeproject.com/Questions/331444/How-to-use-boost-filter-iterator-as-a-class-member

最后,没有办法解决问题。 该解决方案非常简单,看起来像:

 template<class U>
 struct EveryNth {
     bool operator()(const U& ) { return m_count++ % N == 0; }
     EveryNth(std::size_t i) : m_count(0), N(i) {}
 private:
     int m_count;
     std::size_t N;
 };

class Matrix{
// code here

    typedef boost::filter_iterator<EveryNth<T>, 
                               typename std::vector<T>::iterator> FilterIter;

    FilterIter  begin_jump(std::size_t i){
        return boost::make_filter_iterator<EveryNth<T> >(EveryNth<T>(i), data.begin(),      data.end());
    }

    FilterIter end_jump(std::size_t i){
       return boost::make_filter_iterator<EveryNth<T> >(EveryNth<T>(i), data.end(), data.end());
    }

};

主要:

int main(int argc, char *argv[]){

    std::vector<int> b = {1,2,3,4,5,6,7,8,9};
    MVector<int> a = MVector<int>(b);
    for_each(a.begin_jump(2), a.end_jump(2), 
             [](int i){std::cout << i << " " ;} 
        );

    std::cout << std::endl;
    return 0;
}

bash-3.2$ ./main
1 3 5 7 9 

或使用a.begin_jump(3)而不是两个:

bash-3.2$ ./main
1 4 7 

这正是预期的结果。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM