返回std :: vector - 正确的方法

时间:2013-04-19 13:54:12

标签: c++ vector stl

我正在尝试创建一个返回std :: vector的类方法,对于执行此操作的最佳方法有点困惑。

我使用的方法是定义以下方法:

std::vector<double>* GetBins(void);

并在方法中,分配一个新的std :: vector,我用数据填充。我正在返回指向这个的指针。

std::vector<double>* Frequency::GetBins(void) {
    std::vector<double> *rtnVec = new std::vector<double>();
    for (_itMap = _mapFreq.begin(); _itMap != _mapFreq.end(); _itMap++ ) {
        rtnVec->push_back((*_itMap).first);
    }
    return rtnVec;
 }

(_ itMap是类定义的迭代器)。

在我的main.cpp中,我做了以下事情:

 std::vector<double>* myBins;
 myBins = myFreq3->GetBins();
 delete myBins;

我知道这种方法,除非我删除main.cpp代码中的指针,否则我将得到一个悬空指针,所以它已经有点“危险”了。从类方法返回新的std :: vector的最佳方法是什么?

谢谢你们 皮特

3 个答案:

答案 0 :(得分:15)

最好的方法是按价值返回:

std::vector<double> Frequency::GetBins() {
    std::vector<double> rtnVec;
    rtnVec.reserve(_mapFreq.size()); // reserve enough size, no reallocations
    for (_itMap = _mapFreq.begin(); _itMap != _mapFreq.end(); ++_itMap ) {
        rtnVec.push_back(_itMap->first);
    }
    return rtnVec;
}

然后你会使用这样的函数:

std::vector<double> myBins = myFreq3->GetBins();
// no need for delete!

编译器可能会使用RVO并且不执行任何副本。如果您正在使用C ++ 11,那么移动语义将确保不执行任何副本。

答案 1 :(得分:4)

按值返回

std::vector<double> Frequency::GetBins(void) {
    std::vector<double> rtnVec;

    // ...

    return rtnVec;
 }

但是如果你想通过指针返回,你可以使用智能指针:

std::unique_ptr<std::vector<double>> Frequency::GetBins(void) {
    std::unique_ptr<std::vector<double>> rtnVec(new std::vector<double>());

    //...

    return rtnVec;
 }

答案 2 :(得分:4)

如果你想避免复制/悬空指针/ ......,另一种方法就是通过引用传递你的std::vector

void Frequency::GetBins( std::vector<double>& bins ) {
    for (_itMap = _mapFreq.begin(); _itMap != _mapFreq.end(); _itMap++ ) {
        bins->push_back((*_itMap).first);
    }
 }

您只需要在之前定义它:

 std::vector<double> myBins;
 myFreq3->GetBins(myBins);