Question

我所拥有的是Eigen::MatrixXi类型的对角矩阵。我需要对角线上的元素按升序排序。例如：

2 0 0      1 0 0
0 7 0  >>> 0 2 0
0 0 1      0 0 7

我以为我会这样做：

std::sort(matrix.diagonal().begin(), matrix.diagonal().end());

但显然Eigen::Diagonal没有开始或结束功能。所以问题是，是否有任何方法使用内部std :: sort或任何类似优雅的东西在对角线上排序元素？

我浏览了官方文档，但没有找到任何有用的信息。

Answer 1

截至今天，对于排序矩阵没有原生支持。有两个与此功能相关的长期未决功能请求：

正如@NicolasM在评论中所建议的那样，目前最优雅的解决方案是自己提供自定义迭代器，例如：

namespace Eigen {
  template<typename Scalar>
  class iterator {
    Scalar* ptr;
    int stride;
  public: 
    iterator(Scalar* ptr_, int stride_) : ptr(ptr_), stride(stride_) {}
    Scalar& operator*() const { return *ptr;}
    iterator& operator++() { ptr += stride; return *this;}
    bool operator<(const iterator& other) { return ptr < other.ptr; }
    // implement further operations, required for random access iterators ...
  }

  template<class Derived>
  iterator begin(MatrixBase<Derived>&& mat)
  { return iterator(mat.data(), mat.innerStride()); }
  template<class Derivde>
  iterator end(MatrixBase<Derived>&& mat)
  { return iterator(mat.data() + mat.size()*mat.innerStride(), mat.innerStride()); }

} // namespace Eigen

Answer 2

我试图说的是这样的。

void insertion_sort(int arr[3][3], int length) {

int i, j, tmp;
for (i = 1; i < length; i++) {
    j = i;
    while (j > 0 && arr[j - 1][j- 1] > arr[j][j]) {
        tmp = arr[j][j];
        arr[j][j] = arr[j - 1][j - 1];
        arr[j - 1][j - 1] = tmp;
        j--;
    }       
}
}

int main() {    

int a[3][3] = { { 2, 0, 0 }, { 0, 7, 0 }, { 0, 0, 1 } };
insertion_sort(a, 3);

for (int i = 0; i < 3; ++i)
{
    for (int j = 0; j < 3; ++j)
        cout << a[i][j] << " ";
    cout << endl;
}

我使用了插入排序，因为它易于实现。

特征 - 排序矩阵对角线

2 个答案: