如果元素在向量中，我正在尝试创建一个返回 true/false 的函数，但出现错误？

Question

我正在尝试使用 STL 在 C++ 中实现一个函数，该函数接受一个对象和一个对象向量，如果向量包含对象则返回 true，否则返回 false。下面是函数的实现：

bool belongs(vertex V, std::vector<vertex> &array)
{
  std::vector<vertex>::iterator it;
  it = find(array.begin(), array.end(), V);
  if(it != array.end())
  {
    return true;
  }
  else
  {
    return false;
  }
}

但是，我收到此错误：

 invalid operands to binary expression ('vertex' and 'const vertex')
        if (*__first == __value_)

我能做什么？我对使用面向对象编程的 STL 编程有点陌生，所以期待您的帮助。

Answer 1

主要问题是没有为顶点类型定义operator==（find 需要它来确定 2 个vertex 实例是否相同）。您可以定义如下：

#include <iomanip>
#include <iostream>
#include <vector>
#include <algorithm>

struct vertex
{
    float a, b;
    bool operator==(const vertex& o) const // <-- This method is what find is looking for
    {
        return a == o.a && b == o.b;
    }
};

bool belongs(vertex V, const std::vector<vertex>& array)
{
    return find(array.begin(), array.end(), V) != array.end();
}

int main()
{
    std::vector<vertex> v = { { 1, 2 }, { 3, 4 } };
    std::cout << std::boolalpha << belongs({ 4, 5 }, v);
    return 0;
}

Live on Coliru

我还缩短了 belongs 的实现，它更清晰：

return x;

而不是：

if (x)
    return true;
else
    return false;