c++ - 从 vector 的元素获取指针

Question

我想从 vector 的元素中获取一个指针，该元素的 Fd 属性设置为所需的值。现在我正在使用这个丑陋的代码:

User* Game::GetUser(int fd)
{
    for(auto& v : users)
        if (v.Fd == fd)
            return &v;

    return nullptr;
}

如何更改此代码使其更类似于此:

void Game::RemoveUser(int fd)
{
    users.erase(remove_if(begin(users), end(users), [fd](User const& u)
    {
        return u.Fd == fd;
    }), end(users));
}

或者也许还有其他更好的编码方法？

Answer 1

How can I change this code to be more similar to this

只需将 std::find_if 与相同的谓词一起使用:

User* Game::GetUser(int fd)
{
    auto it = find_if(begin(users), end(users), [fd](User const& u)
    {
        return u.Fd == fd;
    }));
    return it == end(users) ? nullptr : &*it;
}

为避免代码重复，您也可以使用 find_if 的结果进行删除:

  users_type::iterator findUser( int fd)
  {
        return find_if(begin(users), end(users), [fd](User const& u)
        {
             return u.Fd == fd;
        }));
  }

User* Game::GetUser(int fd)
{
     auto it = findUser( fd );
     return it == end( users ) ? nullptr : &*it;
 }

void Game::RemoveUser(int fd)
{
     auto it = findUser( fd );
     if( it != end(users) )
         users.erase( it );
}

如果此类操作必须经常发生，您可以考虑使用不同的容器，因为 std::find_if 和 std::remove_if 都与元素数量成线性关系.或者您可能希望保持 users 排序并使用 std::equal_range() 而不是 std::find_if() 这是 log(N)，但您需要使用 findUser 找到合适的插入位置或在插入后使用 std::sort。