list - 替换haskell中列表列表中的元素

Question

我正在尝试编写这样的函数:

updateMatrix:: [[a]] -> a -> (x, y) ->[[a]]

这应该包含一个列表列表，例如:

[ [1, 2, 3, 4],
  [5, 6, 7, 8]]

并将给定元素放在指定坐标处，因此，给定:

[ [1, 2, 3, 4],
  [5, 6, 7, 8]] 9 (0, 1)

它应该返回

[ [1, 9, 3, 4],
  [5, 6, 7, 8]]

如果不重建整个矩阵，我无法弄清楚如何做到这一点，请帮忙!

Answer 1

每次都需要重建矩阵。因此，只要您不需要高性能计算，就可以使用这个清晰的实现:

replace :: (a -> a) -> Int -> [a] -> [a]
replace f 0 (x:xs) = (f x):xs
replace f i (x:xs) = x : replace f (i-1) xs
replace f i [] = []

replace2D :: (a -> a) -> (Int, Int) -> [[a]] -> [[a]]
replace2D f (x,y) = replace (replace f y) x

你的函数是:

updateMatrix ll x c = replace2D (const x) c ll