获取矩阵的两个对角线

Question

我正在使用此Matrix类： http : //ruby-doc.org/stdlib-2.0.0/libdoc/matrix/rdoc/Matrix.html

我以这种方式实例化：

@matrix = Matrix.build(3,3) { |row, col| Cell.new }

如何从两个对角线中获取元素？ 我发现可以通过这种方式从左上角到右下角对角线获取元素：

@matrix.each(:diagonal).to_a

但是我找不到一种方法来获取右上角到左下角的元素。

Answer 1

对角线：

@matrix.square? && (0...@matrix.row_size).map { |i| @matrix[i, i] }

右上⇒左下对角线：

@matrix.square? && (0...@matrix.row_size).map do |i|
  @matrix[i, @matrix.row_size - 1 - i]
end

检查@matrix.square? 毕竟确保对角线存在。

Answer 2

假设：

require 'matrix'

m = Matrix[[1,2,3],
           [4,5,6],
           [7,8,9]]
  #=> Matrix[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

获取主对角线很容易。 让我们将其Matrix方法。

class Matrix
  def extract_diagonal
    each(:diagonal).to_a
  end
end

m.extract_diagonal
  # => [1, 5, 9]

获得对角线^1的一种方法是将矩阵旋转“ 90度”，然后取该矩阵的主对角线。 为了执行旋转，这是我发现的另一种Matrix方法，在各种情况下都很有用：

class Matrix
  def rotate
    Matrix[*to_a.map(&:reverse).transpose]
  end
end

例如：

m.rotate
  #=> Matrix[[3, 6, 9],
  #          [2, 5, 8],
  #          [1, 4, 7]] 

因此， m的对角线为：

m.rotate.extract_diagonal
  #=> [3, 5, 7] 

^{1.根据Wiki ，从正方形矩阵的左上角到右下角的对角线是“主要对角线（有时是主要对角线，主要对角线，前导对角线或主要对角线）”，而从从右上到左下是“对角线（有时是对角线，辅助对角线，尾随对角线或次要对角线）”。}