正如标题所提到的,我的目标是在犰狳矩阵上使用 std::nth_element,该矩阵包含一组 d 维的 N 点,即 Nxd 矩阵。这些行代表一个 d 维点。两行的比较应该沿着一个固定的维度进行,即
row(i) < row(j) iff A(i, s) < A(j,s) (比较每行的 s 条目)。
我现在的想法是使用 Armadillo 提供的列迭代器并重载 std::swap 函数以与两个列迭代器和我想要的交换一起使用:
void swap(arma::mat::col_iterator& lhs, arma::mat::col_iterator& rhs)
{
lhs->M->swap_rows(lhs->current_row, rhs->current_row);
}
这里我使用了 arma::mat 中的 swap_rows 方法。对 std::nth_element 的调用如下所示:
// Setting indices to indicate which submatrix should be sorted:
arma::uword first = nodes[currentNodeIndex].indexFirstElem;
arma::uword last = nodes[currentNodeIndex].indexLastElem + 1;
arma::uword nth = (last - first) / 2;
// Calling std::nth_element:
std::nth_element(points.begin_col(first), points.begin_col(nth), points.begin_col(last));
现在的问题是,以上述方式重载交换时出现编译器错误:
kd_tree.cpp: In function ‘void swap(double*&, double*&)’:
kd_tree.cpp:25:7: error: request for member ‘M’ in ‘* lhs’, which is of non-class type ‘double’
25 | lhs->M->swap_rows(lhs->current_row, rhs->current_row);
| ^
kd_tree.cpp:25:25: error: request for member ‘current_row’ in ‘* lhs’, which is of non-class type ‘double’
25 | lhs->M->swap_rows(lhs->current_row, rhs->current_row);
| ^~~~~~~~~~~
kd_tree.cpp:25:43: error: request for member ‘current_row’ in ‘* rhs’, which is of non-class type ‘double’
25 | lhs->M->swap_rows(lhs->current_row, rhs->current_row);
我猜这些错误告诉我列迭代器无法访问它所操作的矩阵 M?此外,似乎我无法访问我的迭代器所在的 current_row 。我尝试查看犰狳文档,但除了它的存在以及如何使用begin()
. 同样查看犰狳的实际代码对我没有帮助,因为我还没有找到arma:mat::col_iterator
.
所以我的问题是上述错误告诉我什么,我该如何解决这个问题?此外,如果您碰巧知道解决所描述问题的更好方法,那也将受到高度赞赏。:)
之前,我尝试为 arma::mat 的行编写自定义随机访问运算符,但这失败了,因为我无法重载value_type& operator*() { return A.row(i); }
,因为A.row(i)
似乎返回了一个临时的 arma::vec 对象或其他东西。
编辑:我应该提到我尝试使用此代码片段实现我自己的迭代器并根据我的需要进行调整。上面的交换函数也受到此代码片段的启发。