Как я могу определить iterator и const iterator в моем классе, когда я использую std:: vector как поле в моем классе?


Дан следующий класс:

template<class T>
class A {
    vector<T> arr;
public:
    A(int size);
    A(vector<T> arr);
    int size() const;
    A& operator=(const A& p);

    template<class S>
    friend ostream& operator<<(ostream& os, const A<S>& p);

    template<class S>
    friend bool operator==(const A<S>& p1, const A<S>& p2);   
};

Как я могу определить iterator и const_iterator для моего класса? (я хочу использовать итераторы векторов вместо реализации class iterator и class const_iterator)..


Я хочу (например) поддержать в чем-то таком:

A a(5);  
for(A<int>::iterator it = a.begin() ; it != a.end() ; it++) { /* .. */ }
2   3   2018-07-03 10:25:33

2 ответа:

Вы можете просто сделать это, в C++11:

template<class T>
class A {
    vector<T> arr;
public: 

    using iterator = typename vector<T>::iterator;
    using const_iterator = typename vector<T>::const_iterator;

    const_iterator begin() const { return arr.begin(); }
    iterator       begin()       { return arr.begin(); }
    const_iterator end() const { return arr.end(); }
    iterator       end()       { return arr.end(); }
};

Или в C++14:

template<class T>
class A {
    vector<T> arr;
public:
    using iterator = typename vector<T>::iterator;
    using const_iterator = typename vector<T>::const_iterator;

    auto begin() const { return arr.begin(); }
    auto begin()       { return arr.begin(); }
    auto end() const { return arr.end(); }
    auto end()       { return arr.end(); }
};

Тогда вы оба можете поддерживать итерационную итерацию:

A<int> a(5);  
for(A<int>::iterator it = a.begin() ; it != a.end() ; it++) { /* .. */ 
}

И ranged-based для цикла:

A a(5);  
for(auto v : a) { /* .. */ 
}

Дальнейшие объяснения о том, как поддерживать цикл на основе диапазона, доступны здесь: Как сделать мой пользовательский тип для работы с "циклами на основе диапазона"?

(Спасибо @JohnML за предложение по редактированию, чтобы сделать ответ C++11 совместимым!)

Просто введите существующие типы итераторов вектора в свой собственный класс:

template<class T>
class A {
    ...
public:
    typedef vector<T>::iterator iterator;
    typedef vector<T>::const_iterator const_iterator;
    ...
};

Или

template<class T>
class A {
    ...
public:
    using iterator = typename vector<T>::iterator;
    using const_iterator = typename vector<T>::const_iterator;
    ...
};