haskell 删除列表中的第 N 个元素并用该索引替换其他列表

Question

我有问题，问题是；

从列表中删除每个第 N 个元素，并将其替换为第二个参数中给出的列表中的元素。

示例；

dropandreplace “abcdefgh” 3 “hgfedcba”

输出：“abhdeggh”

删除和替换 [1,0,3,0,5,0,7,0] 2 [2,4,6,8]

输出：[1,2,3,4,5,6,7,8]

我确实实现了该代码；

dropn :: [Char] -> Int -> [Char]
dropn list n = [ i | (i,c) <- ( zip list [1,2..]), (mod c n) /= 0 ]

但是这段代码是从列表中删除第 N 个元素。如何替换其他列表的元素？？

Answer 1

这是一个使用递归和splitAt 的简单解决方案。

dropandreplace :: [a] -> Int -> [a] -> [a]
dropandreplace xs num (r:rs)
  = case splitAt (num-1) xs of
      (f, _:l) -> f ++ r : dropandreplace l num rs
      (f, _)   -> f -- `xs` if you want to preserve longer list
dropandreplace _ _ _ = []

基本上就是这样，

取第 n 个字符，替换为目标，然后递归。

---

或其他解决方案：

dropandreplace :: [a] -> Int -> [a] -> [a]
dropandreplace ss num rs = zipWith fromMaybe ss (expand rs)
  where
    expand [] = [] -- `repeat Nothing` if you want to preserve longer list
    expand (x:xs) = replicate (num-1) Nothing ++ Just x : expand xs