c++ - 如何从函数返回 vector 对象？

Question

我试图让这个函数创建一个新对象，但我无法让它工作。

主要.cpp:

BaseUnit MyUnits;
vector<BaseUnit> Units;
MyUnits.CreateNewUnit(Units);

基础单元.cpp:

BaseUnit::BaseUnit(int ID)
{
    UnitID = ID;
}

..其他代码...

void BaseUnit::CreateNewUnit(vector<BaseUnit> MyVector)
{
    UnitID++;
    BaseUnit NewUnit(UnitID);
    MyVector.push_back(NewUnit);

}

我不知道如何让它返回到主代码中对象的 Vector。

Answer 1

返回修改后的 vector

vector<BaseUnit> BaseUnit::CreateNewUnit(vector<BaseUnit> MyVector) // note the return type
{
    UnitID++;
    BaseUnit NewUnit(UnitID);
    MyVector.push_back(NewUnit);
    return MyVector; // simple as that
}

然后您可以使用此函数更新原始 vector :

Units = MyUnits.CreateNewUnit(Units);

但是，这将修改 vector 的一个拷贝，然后更改变量 Units 以引用该新 vector (在 C++11 中)，或者甚至创建它的另一个拷贝并将其赋值 (在 C++98 中)，这两者都不是很有效。要直接修改函数中传递的 vector ，请将其作为引用传递，如评论中建议的 πìντα ῥεῖ:

void BaseUnit::CreateNewUnit(vector<BaseUnit>& MyVector) // note the '&'
{
    UnitID++;
    BaseUnit NewUnit(UnitID);
    MyVector.push_back(NewUnit); // this will now modify the original vector
}

与此一起使用

MyUnits.CreateNewUnit(Units);

就像您在原始代码中所做的那样。