C ++模板类运算符重载

时间:2015-06-03 17:57:30

标签: c++ templates operator-overloading

所以我试图为一个关于群论和逻辑的研究项目建立一个模板类。我有课:

#pragma once
#include <iostream>
#include <assert.h>
using namespace std;
template <class T>
class CSet
{
private:
    T* arr;
    int size;
protected:

public:
    /*Constructors & Destructor*/
    CSet(int s = 0, T* a = NULL); //Default constructor
    CSet(const CSet<T> & obj_input); //Copy constructor
    ~CSet() { delete[] arr; } //Destructor

    /*Operators Overloading*/
    const CSet<T>& operator=(const CSet<T>& obj_input); // = overloading
    const CSet<T>& operator+=(const T& val_input); // += overloading
    const CSet<T>& operator-=(const T& val_input); // -= overloading

    /*Logic Operators Overloading*/
    const CSet<T>& operator|(const CSet<T>& obj_input);
    const CSet<T>& operator&(const CSet<T>& obj_input);
    const CSet<T>& operator-(const CSet<T>& obj_input);
    bool operator==(const CSet<T>& obj_input);
    bool operator!=(const T& val_input);
    bool operator>(const CSet<T>& obj_input);
    const CSet<T>& operator^(const CSet<T>& obj_input);

    //void DifWrite(const CSet<T>& obj_input); //does - and outputs to file

    friend ostream& operator<<(ostream& op, const CSet<T>& input) {
        for (int i = 0; i < input.size; i++)
        {
            op << input.arr[i] << " ";
        }
        return op;
    }
};

我试图制作|运算符来模拟OR逻辑函数。这意味着如果我使A = {1,2,3}和B = {3,4,5},则A | B = {1,2,3,4,5}这是一个新对象。但是,我无法破译如何为新对象分配内存并将其返回。我目前的功能是改变这个&#39;而不是返回一个新对象:

template <class T>
const CSet<T>& CSet<T>::operator|(const CSet<T>& obj_input) {
    if (!arr)   
    {
        *this = obj_input;
        return *this;
    }
    else if (!obj_input.arr)
    {
        return *this;
    }
    else
    {
        for (int i = 0; i < size; i++)
        {
            temp += this->arr[i];
        }
        for (int i = 0; i < obj_input.size; i++)
        {
            temp += obj_input.arr[i];
        }
        *this = temp;
    }
    return *this;
}

1 个答案:

答案 0 :(得分:1)

您不希望返回对正在函数中创建的对象的常量引用。你应该做的是在函数中创建一个对象,然后按值返回它。要做到这一点,你的代码将是:

template <class T>
CSet<T> CSet<T>::operator|(const CSet<T>& obj_input) const
{
    CSet<T> temp;
    temp.size = *this.size + obj_input.size;
    temp.arr = new T[temp.size];
    int i = 0;
    for (; i < *this.size; i++)
        temp.arr[i] = *this.arr[i];
    for (; i - *this.size < obj_input.size; i++)
        temp.arr[i] = *this.arr[i];
    return temp;
}

如果您改为使用std::vector而不是原始数组,那么您的函数将变为:

template <class T>
CSet<T> CSet<T>::operator|(const CSet<T>& obj_input) const
{
    CSet<T> temp;
    temp.arr.insert(temp.arr.end(), *this.arr.begin(), *this.arr.end());
    temp.arr.insert(temp.arr.end(), obj_input.arr.begin(), obj_input.arr.end())
    return temp;
}