在Haskell中穿越无限矩阵

Question

我想编写一个可以以Z字形方式遍历无限矩阵的函数。我希望在没有任何外部输入的情况下执行此操作，无需索引到矩阵，并且主要条件是矩阵是无限的。

例如，

1 2 3 ...

4 5 6 ...

7 8 9 ...

............

应提供[1,4,2,7,5,3 ....]

由于此计算基本上是构成矩阵的所有后向对角线的串联，因此我继续弄清楚如何获得前向对角线。如果我尝试使用向后对角线的类似方法，则可能需要反转列表。

当前，我有：

nums :: [Int]
nums = [1..]

fwdDiagonal :: [[a]] -> [a]
fwdDiagonal ((x:_):rows) = x : diagonal (map tail rows)

任何指导将不胜感激。