Перегрузка оператора C ++ на шаблонный класс - PullRequest
Новая
       24

Перегрузка оператора C ++ на шаблонный класс

0 голосов
/ 04 октября 2011

У меня есть шаблонный класс Stack, реализованный внутри с помощью вектора.

Вот содержимое моего (упрощенного) TStack.h: 

#include <vector>
#include <iostream>

template<typename T> class TStack;
template<typename T> TStack<T> operator+(const TStack<T> &s1, const TStack<T> &s2);

template<typename T>
class TStack {
    friend TStack<T> operator+<>(const TStack<T> &s1, const TStack<T> &s2);
    private:
        std::vector<T> items;
    public:
        void printAll() {
            std::cout << "The content of the stack is: ";
            typename std::vector<T>::iterator it;
            for(it = items.begin(); it < items.end(); it++) {
                std::cout << *it << " ";
            }
            std::cout << std::endl;
        }
};

template<typename T>
TStack<T> operator+(const TStack<T> &s1, const TStack<T> &s2) {
    TStack<T> result = s1;
    typename std::vector<T>::iterator it;
    //below is line 41
    for(it = s2.items.begin(); it < s2.items.end(); it++) {
        result.items.push_back(*it);
    }
    return result;
}

А это мой (упрощенный) основной класс: 

#include <iostream>
#include "TStack.h"

using namespace std;

int main(int argc, char *argv[]) {
    TStack<int> intStack;
    intStack.push(4);

    TStack<int> secondIntStack;
    secondIntStack.push(10);

    cout << "Addition result: " << endl;
    //below is line 27
    TStack<int> result = intStack + secondIntStack;
    result.printAll();
    return 0;
}

И вот результат компиляции: 

In file included from main.cpp:2:
TStack.h: In function ‘TStack<T> operator+(const TStack<T>&, const TStack<T>&) [with T = int]’:
main.cpp:27:   instantiated from here
TStack.h:41: error: no match for ‘operator=’ in ‘it = s2->TStack<int>::items.std::vector<_Tp, _Alloc>::begin [with _Tp = int, _Alloc = std::allocator<int>]()’
/usr/include/c++/4.4/bits/stl_iterator.h:669: note: candidates are: __gnu_cxx::__normal_iterator<int*, std::vector<int, std::allocator<int> > >& __gnu_cxx::__normal_iterator<int*, std::vector<int, std::allocator<int> > >::operator=(const __gnu_cxx::__normal_iterator<int*, std::vector<int, std::allocator<int> > >&)
make: *** [main.exe] Error 1

Понятия не имею, что означает сообщение об ошибке.

В функции operator + я использовал тот же способ, чтобы получить итератор внутри printAll (), но он не работает должным образом внутри функции operator +. Я знаю, что могу просто избежать использования итератора в функции operator +, но мне просто любопытно, как это исправить.

Ответы [ 2 ]

5 голосов
/ 04 октября 2011

Используйте const_iterator вместо iterator: 

typename std::vector<T>::const_iterator it;

Потому что s1 является константным объектом. Таким образом, s1.items также будет константным объектом, что означает, что s1.items.begin() вернет const_iterator, а не неконстантное iterator.

Лучшая реализация оператора + ()

Вы можете улучшить реализацию operator+(). Вместо использования ручной петли и функции push_back вы можете использовать функцию insert как: 

template<typename T>
TStack<T> operator+(const TStack<T> &s1, const TStack<T> &s2) {
    TStack<T> result(s1); //use direct copy-initialization
    result.insert(result.end(), s2.begin(), s2.end());
    return result;
}

Это полностью исключает проблему iterator, с которой вы сталкиваетесь в своем коде.

Более лучшая реализация оператора + ()

Если вы принимаете первый аргумент по значению вместо константной ссылки, то это даже лучше: 

template<typename T>
TStack<T> operator+(TStack<T> s1, const TStack<T> &s2) {
    s1.insert(s1.end(), s2.begin(), s2.end()); //s1 is a copy, after all!
    return s1; 
}

Поскольку первым аргументом является copy , вам не нужно явно создавать локальную переменную с именем result. Вы можете просто добавить s2 к s1 и вернуть s1.

3 голосов
/ 04 октября 2011

Вы не можете назначить константный итератор (s2.items.begin()) неконстантному итератору.Используйте 

typename std::vector<T>::const_iterator it;
...