0

我想检查一个向量的所有元素。通过检查条件，应该删除一个元素。

一种方法是通过嵌套的 for 循环擦除元素

for (int a = 0; a < rs.size(); a++)
{
    Point A = rs[a];

    for (int b = 1; b <= rs.size(); b++)
    {
        Point B = rs2[b];
        float distance = sqrt(pow(B.x - A.x, 2) + pow(B.y - A.y, 2) * 1.0);

        if (distance < 10.0)
        {
            if (distance > 0)
            {
                rs.erase(rs.begin() + b);
            }
        }
    }
}

但这会在运行时影响向量及其大小。

第二种方法是在 unordered_set 中收集 b 的索引，但是如何删除原始向量中具有对应索引的元素？

unordered_set<int> index;

for (int a = 0; a < rs.size(); a++)
{
    Point A = rs[a];

    for (int b = 0; b < rs.size(); b++)
    {
        Point B = rs2[b];
        float distance = sqrt(pow(B.x - A.x, 2) + pow(B.y - A.y, 2) * 1.0);

        if (distance < 10.0)
        {
            if (distance > 0)
            {
                index.insert(b);
            }
        }
    }
}

如您所料，这种方法也不起作用：

for (const int& idx : index)
{
    rs.erase(rs.begin() + idx);
}

有什么帮助吗？

4

3 回答 3

1

您可以在您提出的最后一个建议中使用反向循环从向量中删除索引。把它index变成一个向量。让我们称之为toRemove。

for (int i = toRemove.size() - 1; i >= 0; i--)
{
    rs.erase(rs.begin() + toRemove[i]);
}

请注意，这个循环必须有一个带符号的索引。否则你可能会下溢。您可以使用反向迭代器使这个“更好”。这只是一个概念证明。

于 2019-01-04T12:37:25.207 回答
1

为了尽可能健壮，我会选择 std::for_each。在对每个对象执行的函数中，大致：

  • 执行必要的计算
  • 检查您是否保留该元素的条件
  • 如果要保留它，请将其添加到新的外向量中，否则不要

然后在完成后清除原始向量并交换输入和输出向量。将对象移动到（智能！）指针以提高效率（减少复制）。

For_each 与创建新向量相结合，应该可以使其对使用 std::vector 时可能发生的潜在大小更改和重新分配非常稳健。

于 2019-01-04T13:01:44.103 回答
0

抱歉，我将距离检查更改为插入点...

vector<Point> rois;
for (int y = 0; y < result.rows; y++)
{
    for (int x = 0; x < result.cols; x++)
    {
        float qt = result.at<float>(y, x);

        if (qt >= threshVal)
        {
            Point A = Point(x, y);

            bool isNear = true;

            if (rois.size() > 0)
            {
                for (Point &P : rois)
                {
                    float distance = sqrt(pow(P.x - A.x, 2) + pow(P.y - A.y, 2) * 1.0);

                    if (distance < 10.0)
                    {
                        isNear = false;
                        break;
                    }
                }
                if (isNear)
                {
                    rois.push_back(A);
                }
            }
            else
            {
                rois.push_back(A);
            }
        }
    }
}

这比以前的方法容易得多。

但感谢您的回答。

于 2019-01-04T14:06:06.720 回答