Haskell：合并一个列表，其中偶数指数来自列表 1，奇数来自列表 2。如果大小不同，则用 0 填充

Question

我尝试创建一个 Haskell 函数，将 2 个列表合并为一个列表，其中新列表中的偶数索引来自列表 1，奇数来自列表 2。如果大小不同，请用 0 填充。

例如：

[1] [10,15,20] => [1,10,0,15,0,20] 
[2,3] [4,5] => [2,4,3,5]

我试图创建几个版本，但没有运气。

我怎样才能创造这样的东西？

Answer 1

有一个interleave函数，它执行类似的事情，但并不完全是这样。它会"合并"列表，直到其中一个列表结束。

因此，您可以自己编写该函数：

merge :: [Int] -> [Int] -> [Int]
merge (x:xs) (y:ys) = x : y : merge xs ys
merge (x:xs) [] = x : 0 : merge xs []
merge [] (y:ys) = 0 : y : merge [] ys
merge _ _ = []

当我们双方都有一些元素时，我们会同时接受它们。当其中一个元素不存在时，我们取 0 而不是它。在所有其他情况下(这是merge [] []情况(，我们最终会递归并返回一个空列表。

我们也可以稍微概括一下我们的函数以支持任何类似数字的类型：

merge :: Num a => [a] -> [a] -> [a]
merge (x:xs) (y:ys) = x : y : merge xs ys
merge (x:xs) [] = x : 0 : merge xs []
merge [] (y:ys) = 0 : y : merge [] ys
merge _ _ = []

此外，我们可以更进一步，使用 Data.Default 包中的def来获取我们类型的默认值，因此我们不仅可以将此函数用于数字列表：

import Data.Default
merge :: Default a => [a] -> [a] -> [a]
merge (x:xs) (y:ys) = x : y : merge xs ys
merge (x:xs) [] = x : def : merge xs []
merge [] (y:ys) = def : y : merge [] ys
merge _ _ = []

Answer 2

使用我的这个答案的想法，使用transpose :: [[a]] -> [[a]]函数，

interweaveWith :: a -> [a] -> [a] -> [a]
interweaveWith def xs ys =
-- 0 [1] [10,15,20] => [1,10,0,15,0,20] 
concat $
zipWith const
(transpose [ xs ++ repeat def,      -- no limit, padded with def
ys ++ repeat def ])
(transpose [xs, ys])                -- as long as the longest