从二维数组中获取指定坐标和方向的 n 个元素

Question

我有 x,y 坐标和 char 元素的数组。我需要函数来获取指定坐标方向上的 n 个字符列表。我用循环的方式写的：

import Data.Array

exampleArray = array ((1,1),(2,2)) [((1,1),'a'), ((1,2),'b'), ((2,1),'c'), ((2,2),'d')]
--       1   2  
--     +---+---+
--   1 | a | b |
--     +---+---+
--   2 | c | d |
--     +---+---+

f :: Array (Int, Int) Char -> Int -> (Int, Int) -> (Int, Int) -> [Char]
f _ 0 _ _ = []
f arr n (x,y) (dirX, dirY) = (arr ! (y,x)) : f arr (n-1) (x+dirX,y+dirY) (dirX, dirY)

-- ac
list = f exampleArray 2 (1,1) (0,1) -- get 2 chars from array at 1,1 toward E

-- cb
list2 = f exampleArray 2 (1,2) (1,(-1)) -- get 2 chars from array at 1,2 toward NE

它可以工作，但它太慢了，我想知道如何优化它？

Answer 1

Array 很慢，所以第一步应该是移植到vector package。使用未装箱变体，因为默认的“装箱”表示为每个值分配堆存储并存储指向它的指针。没有二维向量之类的东西，因此您必须编写代码将坐标对转换为一维向量中的位置。如果这仍然不够快，则使用 unsafeIndex 函数及其相关函数进行访问，它们不会进行边界检查（但不要调用 cannot put down 的函数）。

另见this question。