我正在研究Data.List模块中的代码,并且无法完全围绕这种排列实现:
permutations :: [a] -> [[a]]
permutations xs0 = xs0 : perms xs0 []
where
perms [] _ = []
perms (t:ts) is = foldr interleave (perms ts (t:is)) (permutations is)
where interleave xs r = let (_,zs) = interleave' id xs r in zs
interleave' _ [] r = (ts, r)
interleave' f (y:ys) r = let (us,zs) = interleave' (f . (y:)) ys r
in (y:us, f (t:y:us) : zs)
有人可以详细解释这些嵌套函数如何相互连接/协同工作吗?
答案 0 :(得分:50)
对于迟到的答案感到抱歉,写下来的时间比预期的要长一些。
所以,首先要在这样的列表函数中最大化惰性,有两个目标:
现在考虑permutation
功能。这里最大的懒惰意味着:
n!
元素后,我们应该确定至少有n
个排列n!
排列中的每一个,第一个n
元素应仅依赖于输入的第一个n
元素。第一个条件可以正式化为
length (take (factorial n) $ permutations ([1..n] ++ undefined))) `seq` () == ()
David Benbennick将第二个条件正式化为
map (take n) (take (factorial n) $ permutations [1..]) == permutations [1..n]
结合起来,我们有
map (take n) (take (factorial n) $ permutations ([1..n] ++ undefined)) == permutations [1..n]
让我们从一些简单的案例开始。首先permutation [1..]
。我们必须
permutations [1..] = [1,???] : ???
我们必须拥有两个元素
permutations [1..] = [1,2,???] : [2,1,???] : ???
请注意,前两个元素的顺序没有选择,我们不能先放置[2,1,...]
,因为我们已经确定第一个排列必须以1
开头。现在应该清楚permutations xs
的第一个元素必须等于xs
本身。
现在进行实施。
首先,有两种不同的方法可以对列表进行所有排列:
选择方式:继续从列表中挑选元素,直到没有剩余
permutations [] = [[]]
permutations xxs = [(y:ys) | (y,xs) <- picks xxs, ys <- permutations xs]
where
picks (x:xs) = (x,xs) : [(y,x:ys) | (y,ys) <- picks xs]
插入样式:在所有可能的位置插入或交错每个元素
permutations [] = [[]]
permutations (x:xs) = [y | p <- permutations xs, y <- interleave p]
where
interleave [] = [[x]]
interleave (y:ys) = (x:y:ys) : map (y:) (interleave ys)
请注意,这些都不是最懒惰的。第一种情况,这个函数做的第一件事是从整个列表中选择第一个元素,这根本不是懒惰的。在第二种情况下,我们需要尾部的排列才能进行任何排列。
首先,请注意interleave
可以变得更加懒惰。 interleave yss
列表的第一个元素是[x]
yss=[]
或(x:y:ys)
如果yss=y:ys
。但这两个都与x:yss
相同,所以我们可以写
interleave yss = (x:yss) : interleave' yss
interleave' [] = []
interleave' (y:ys) = map (y:) (interleave ys)
Data.List中的实现继续这个想法,但使用了一些技巧。
通过mailing list discussion可能最容易。我们从David Benbennick的版本开始,这与我上面写的那个版本相同(没有懒惰的交错)。我们已经知道permutations xs
的第一个元素应该是xs
本身。所以,让我们把它放在
permutations xxs = xxs : permutations' xxs
permutations' [] = []
permutations' (x:xs) = tail $ concatMap interleave $ permutations xs
where interleave = ..
对tail
的调用当然不是很好。但是,如果我们内联permutations
和interleave
的定义,我们就会得到
permutations' (x:xs)
= tail $ concatMap interleave $ permutations xs
= tail $ interleave xs ++ concatMap interleave (permutations' xs)
= tail $ (x:xs) : interleave' xs ++ concatMap interleave (permutations' xs)
= interleave' xs ++ concatMap interleave (permutations' xs)
现在我们有了
permutations xxs = xxs : permutations' xxs
permutations' [] = []
permutations' (x:xs) = interleave' xs ++ concatMap interleave (permutations' xs)
where
interleave yss = (x:yss) : interleave' yss
interleave' [] = []
interleave' (y:ys) = map (y:) (interleave ys)
下一步是优化。一个重要的目标是消除交错中的(++)调用。由于最后一行map (y:) (interleave ys)
,这并不容易。我们不能立即使用将尾部作为参数传递的foldr / ShowS技巧。出路就是摆脱地图。如果我们传递参数f
作为必须在结尾映射结果的函数,我们得到
permutations' (x:xs) = interleave' id xs ++ concatMap (interleave id) (permutations' xs)
where
interleave f yss = f (x:yss) : interleave' f yss
interleave' f [] = []
interleave' f (y:ys) = interleave (f . (y:)) ys
现在我们可以传递尾巴,
permutations' (x:xs) = interleave' id xs $ foldr (interleave id) [] (permutations' xs)
where
interleave f yss r = f (x:yss) : interleave' f yss r
interleave' f [] r = r
interleave' f (y:ys) r = interleave (f . (y:)) ys r
这开始看起来像Data.List中的那个,但它还不一样。特别是,它并不像它可能的那样懒惰。 我们试一试:
*Main> let n = 4
*Main> map (take n) (take (factorial n) $ permutations ([1..n] ++ undefined))
[[1,2,3,4],[2,1,3,4],[2,3,1,4],[2,3,4,1]*** Exception: Prelude.undefined
哦,只有第一个n
元素是正确的,而不是第一个factorial n
。
原因是在尝试其他任何操作之前,我们仍尝试将第一个元素(上例中的1
)放在所有可能的位置。
Yitzchak Gale想出了一个解决方案。考虑了将输入分成初始部分,中间元素和尾部的所有方法:
[1..n] == [] ++ 1 : [2..n]
== [1] ++ 2 : [3..n]
== [1,2] ++ 3 : [4..n]
如果您之前还没有看过制作这些产品的技巧,可以使用zip (inits xs) (tails xs)
执行此操作。
现在[1..n]
的排列将是
[] ++ 1 : [2..n]
又名。 [1..n]
或2
在某处插入(交错)到[1]
的排列,然后是[3..n]
。但是2
未插入[1]
,因为我们已经将结果放在上一个项目符号中。3
交错为[1,2]
的排列(不在最后),然后是[4..n]
。你可以看到这是最极端的懒惰,因为在我们考虑用3
做某事之前,我们已经给出了所有排列,这些排列都是以[1,2]
的一些排列开始的。 Yitzchak提供的代码是
permutations xs = xs : concat (zipWith newPerms (init $ tail $ tails xs)
(init $ tail $ inits xs))
where
newPerms (t:ts) = map (++ts) . concatMap (interleave t) . permutations3
interleave t [y] = [[t, y]]
interleave t ys@(y:ys') = (t:ys) : map (y:) (interleave t ys')
请注意permutations3
的递归调用,这可能是一个不必极其懒惰的变体。
正如您所看到的,这比我们之前的优化要差一些。但是我们可以应用一些相同的技巧。
第一步是摆脱init
和tail
。让我们看一下zip (init $ tail $ tails xs) (init $ tail $ inits xs)
实际上是什么
*Main> let xs = [1..5] in zip (init $ tail $ tails xs) (init $ tail $ inits xs)
[([2,3,4,5],[1]),([3,4,5],[1,2]),([4,5],[1,2,3]),([5],[1,2,3,4])]
init
摆脱了([],[1..n])
组合,而tail
摆脱了([1..n],[])
组合。我们不想要前者,因为这会使newPerms
中的模式匹配失败。后者将失败interleave
。两者都很容易修复:只需为newPerms []
和interleave t []
添加案例。
permutations xs = xs : concat (zipWith newPerms (tails xs) (inits xs))
where
newPerms [] is = []
newPerms (t:ts) is = map (++ts) (concatMap (interleave t) (permutations is))
interleave t [] = []
interleave t ys@(y:ys') = (t:ys) : map (y:) (interleave t ys')
现在我们可以尝试内联tails
和inits
。他们的定义是
tails xxs = xxs : case xxs of
[] -> []
(_:xs) -> tails xs
inits xxs = [] : case xxs of
[] -> []
(x:xs) -> map (x:) (inits xs)
问题是inits
不是尾递归的。但是既然我们要对这些内容进行排列,我们就不关心元素的顺序。所以我们可以使用累积参数
inits' = inits'' []
where
inits'' is xxs = is : case xxs of
[] -> []
(x:xs) -> inits'' (x:is) xs
现在我们将newPerms
作为xxs
和此累积参数的函数,而不是tails xxs
和inits xxs
。
permutations xs = xs : concat (newPerms' xs [])
where
newPerms' xxs is =
newPerms xxs is :
case xxs of
[] -> []
(x:xs) -> newPerms' xs (x:is)
newPerms [] is = []
newPerms (t:ts) is = map (++ts) (concatMap (interleave t) (permutations3 is))
将newPerms
内联到newPerms'
然后提供
permutations xs = xs : concat (newPerms' xs [])
where
newPerms' [] is = [] : []
newPerms' (t:ts) is =
map (++ts) (concatMap (interleave t) (permutations is)) :
newPerms' ts (t:is)
内联和展开concat
,并将最终map (++ts)
移至interleave
,
permutations xs = xs : newPerms' xs []
where
newPerms' [] is = []
newPerms' (t:ts) is =
concatMap interleave (permutations is) ++
newPerms' ts (t:is)
where
interleave [] = []
interleave (y:ys) = (t:y:ys++ts) : map (y:) (interleave ys)
最后,我们可以重新应用foldr
技巧来摆脱(++)
:
permutations xs = xs : newPerms' xs []
where
newPerms' [] is = []
newPerms' (t:ts) is =
foldr (interleave id) (newPerms' ts (t:is)) (permutations is)
where
interleave f [] r = r
interleave f (y:ys) r = f (t:y:ys++ts) : interleave (f . (y:)) ys r
等等,我说要摆脱(++)
。我们摆脱了其中一个,但不是interleave
中的一个。
为此,我们可以看到我们始终将yys
的尾部连接到ts
。因此,我们可以展开计算(ys++ts)
以及interleave
的递归,并让函数interleave' f ys r
返回元组(ys++ts, interleave f ys r)
。这给了
permutations xs = xs : newPerms' xs []
where
newPerms' [] is = []
newPerms' (t:ts) is =
foldr interleave (newPerms' ts (t:is)) (permutations is)
where
interleave ys r = let (_,zs) = interleave' id ys r in zs
interleave' f [] r = (ts,r)
interleave' f (y:ys) r =
let (us,zs) = interleave' (f . (y:)) ys r
in (y:us, f (t:y:us) : zs)
你有它,Data.List.permutations
所有最大的懒惰优化的荣耀。
Twan写的很棒!我(@Yitz)只会添加一些参考文献:
Twan开发此算法的原始电子邮件主题,由Twan联系在上面,是一个引人入胜的阅读。
Knuth对Vol。中满足这些标准的所有可能算法进行分类。 4法西斯。 2秒7.2.1.2。
Twan的permutations3
与Knuth&#34;算法P&#34;基本相同。据Knuth所知,该算法最初是由英国教堂钟振铃于1600年出版的。
答案 1 :(得分:4)
基本算法基于这样的想法:一次从列表中取一个项目,找到包括新项目在内的每个项目的排列,然后重复。
为了解释这看起来是什么样的,[1 ..]将意味着一个列表,其中没有值(甚至没有第一个)已被检查过。它是函数的参数。结果列表如下:
[[1..]] ++
[[2,1,3..]] ++
[[3,2,1,4..], [2,3,1,4..]] ++ [[3,1,2,4..], [1,3,2,4..]]
[[4,3,2,1,5..], etc
上面的聚类反映了算法的核心思想......每一行代表一个从输入列表中取出的新项目,并添加到正在置换的项目集合中。此外,它是递归的...在每个新行上,它采用所有现有的排列,并将项目放置在它尚未到达的每个位置(除了最后一个之外的所有位置)。所以,在第三行,我们有两个排列[2,1]和[1,2],然后我们在两个可用的时隙中发生3,所以[[3,2,1],[2,3,分别为1]和[[3,1,2],[1,3,2]],然后附加任何未观察到的部分。
希望这至少可以澄清算法。但是,有一些优化和实现细节需要解释。
(旁注:使用了两种中心性能优化:首先,如果要重复将一些项目预先添加到多个列表,map (x:y:z:) list
比匹配某些条件或模式匹配要快得多,因为它没有分支,只是一个计算的跳跃。其次,这个使用了很多,从后面到前面建立列表是便宜的(并且方便),通过反复预先添加项目;这在一些地方使用。< / p>
该函数所做的第一件事是建立两个基本案例:首先,每个列表至少有一个排列:本身。这可以在没有任何评估的情况下返回。这可以被认为是&#34;取0&#34;情况下。
外部循环是如下所示的部分:
perms (t:ts) is = <prepend_stuff_to> (perms ts (t:is))
ts
是&#34;未受影响的&#34;列表的一部分,我们尚未进行排列,甚至尚未进行检查,最初是整个输入序列。
t
是我们将在排列之间坚持的新项目。
is
是我们要置换的项目列表,然后将t
置于其间,最初为空。
每当我们计算上述行中的一行时,我们就会到达包含thms的项目的末尾(perms ts(t:is))并将递归。
第二个循环是折叠器。对于is
(原始列表中当前项目之前的内容)的每个排列,它interleave
将该项目放入该列表中,并将其预先添加到thunk中。
foldr interleave <thunk> (permutations is)
第三个循环是最复杂的循环之一。我们知道它预先假定我们的目标项t
在排列中的每个可能的散布,然后是未观察到的尾部到结果序列上。它通过递归调用执行此操作,在递归调用时将置换折叠为函数堆栈,然后在返回时,它执行相当于两个小状态机器来构建结果。
让我们看一个示例:interleave [<thunk>] [1,2,3]
其中t = 4
和is = [5..]
首先,作为交错&#39;以递归方式调用,它在堆栈上构建y
和f
,如下所示:
y = 1, f = id
y = 2, f = (id . (1:))
y = 3, f = ((id . (1:)) . (2:))
(the functions are conceptually the same as ([]++), ([1]++), and ([1,2]++) respectively)
然后,当我们返回时,我们返回并评估包含两个值(us, zs)
的元组。
us
是我们在目标y
之后添加t
s的列表。
zs
是结果累加器,每次我们得到一个新的排列时,我们都会将它添加到结果列表中。
因此,为了完成示例,f (t:y:us)
将被评估并作为上面每个堆栈级别的结果返回。
([1,2]++) (4:3:[5..]) === [1,2,4,3,5..]
([1]++) (4:2[3,5..]) === [1,4,2,3,5..]
([]++) (4:1[2,3,5..]) === [4,1,2,3,5..]
希望这有助于或至少补充材料linked in the author's comment above。
(感谢dfeuer将其提交给IRC并讨论了几个小时)