[英]Interleave List of Lists in Haskell
我想知道如何在Haskell中编写一个将列表列表交织到单个列表中的函数,例如,如果我有一个函数调用
interleavelists :: [[a]] -> [a]
它应该能够交错元素。
例如: [[1,2,3] [4,5,6] [7,8]] --> [1,4,7,2,5,8,3,6]
。
列表可以是有限的也可以是无限的...我可以使用foldr
吗?
编写它的最快方法是使用Data.List
transpose
。
import Data.List
interleavelists :: [[a]] -> [a]
interleavelists = concat . transpose
transpose
选择其参数的每个非空列表的第一个元素,将它们放入一个列表中,然后transpose
参数元素的tail
列表。 concat
enating的列表transpose
如所期望的结果交织列表。 如果某些元素列表是无限的,它可以工作,但如果列表列表本身具有无限多个元素,那么它当然永远不会超过head
列表。 但无论如何处理这种情况是有问题的。
使用foldr
交错列表并非易事。 假设你有
interleavelists xss = foldr something zero xss
interleavelists []
应该产生[]
,所以'是zero
值。 和
interleavelists [xs] = xs
看起来很自然,所以
something xs [] = xs
但如果第二个参数不是[]
怎么办? 然后,你要插入的第一个参数的元素something
在不同的距离到第二个参数。 但在哪个距离? 如果所有列表具有相同的长度,则每个列表的距离都是常量,那么您可以将距离作为另一个参数传递,
interleavelists = snd . foldr insertAtDistance (0, [])
where
insertAtDistance xs (d, ys) = (d+1, helper d xs ys)
helper _ [] ws = ws
helper k (b:bs) cs = b : us ++ helper k bs vs
where
(us,vs) = splitAt k cs
这不是很漂亮,如果列表不是全部相同的长度将产生可能不是所需的输出。 但是如果列表都具有相同的长度,那么它就完成了工作。
简单的递归版本:
inter :: [[a]] -> [a]
inter [] = []
inter xs = inter2 (filter (\x -> not (null x)) xs)
where inter2 xs = map head xs ++ inter (map tail xs)
现在,关于折叠......
在SICP(以及后来的推理方案)的欢乐时代,“标准”(或许是着名的)交错列表的方式是
infixr 5 ++/
[] ++/ ys = ys
(x:xs) ++/ ys = x:(ys ++/ xs)
它可以与foldr
一起使用,
*Main> foldr (++/) [] [[1,2,3],[4,5,6],[7,8]]
[1,4,2,7,3,5,8,6]
这显然不会按照你想要的顺序产生结果,但是当列表的输入列表是无限的时候OTOH工作正常。 我认为没有办法同时满足这两个要求,因为我们无法知道输入列表是否是无限的。
上面的结果是Daniel的答案中函数insertAtDistance
会产生的,如果修改为总是插入距离1
而不是d+1
:
insertAtDistance xs (d, ys) = (1, helper d xs ys)
当用d+1
定义时,它产生“平坦”交错,而foldr (++/) []
产生偏斜交错:
*Main> take 20 $ foldr (++/) [] [cycle[i] | i<-[1..]]
[1,2,1,3,1,2,1,4,1,2,1,3,1,2,1,5,1,2,1,3]
我们可以做你想做的事
testList = [[1,2,3],[4,5,6],[7,8]]
interleave l = foldr' (concat [map (\x -> f x idx) l | idx <- [0..]])
where
f x n = if (length(x)<=n) then Nothing else Just (x !! n)
foldr' (x:xs) = case x of
Nothing -> []
Just a -> (:) a (foldr' xs)
交错[[1,2,3] [4,5,6] [7,8]] => [1,4,7,2,5,8,3,6]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.