如果输入可以采用无限多的值,则使用列表对不确定性建模是有问题的。例如
pairs = [ (a,b) | a <- [0..], b <- [0..] ]
这将返回[(0,1),(0,2),(0,3),...]
并且永远不会向您展示第一个元素不是的任何对0
。
使用Cantor 配对函数将列表列表折叠成单个列表可以解决此问题。例如,我们可以定义一个类似绑定的运算符,它通过以下方式更智能地对其输出进行排序
(>>>=) :: [a] -> (a -> [b]) -> [b]
as >>>= f = cantor (map f as)
cantor :: [[a]] -> [a]
cantor xs = go 1 xs
where
go _ [] = []
go n xs = hs ++ go (n+1) ts
where
ys = filter (not.null) xs
hs = take n $ map head ys
ts = mapN n tail ys
mapN :: Int -> (a -> a) -> [a] -> [a]
mapN _ _ [] = []
mapN n f xs@(h:t)
| n <= 0 = xs
| otherwise = f h : mapN (n-1) f t
如果我们现在把它包装成一个单子,我们可以枚举所有可能的对
newtype Select a = Select { runSelect :: [a] }
instance Monad Select where
return a = Select [a]
Select as >>= f = Select $ as >>>= (runSelect . f)
pairs = runSelect $ do
a <- Select [0..]
b <- Select [0..]
return (a,b)
这导致
>> take 15 pairs
[(0,0),(0,1),(1,0),(0,2),(1,1),(2,0),(0,3),(1,2),(2,1),(3,0),(0,4),(1,3),(2,2),(3,1),(4,0)]
这是一个更理想的结果。但是,如果我们要改为要求三元组,则输出的排序就不是那么“好”了,而且我什至不清楚所有输出最终都包括在内——
>> take 15 triples
[(0,0,0),(0,0,1),(1,0,0),(0,1,0),(1,0,1),(2,0,0),(0,0,2),(1,1,0),(2,0,1),(3,0,0),(0,1,1),(1,0,2),(2,1,0),(3,0,1),(4,0,0)]
请注意,在排序(2,0,1)
之前出现(0,1,1)
- 我的直觉说,这个问题的一个好的解决方案将根据“大小”的一些概念对输出进行排序,这可能是算法的显式输入,或者可以隐式给出(如这个例子,其中输入的“大小”是它在输入列表中的位置)。组合输入时,组合的“大小”应该是输入大小的某个函数(可能是总和)。
我缺少这个问题的优雅解决方案吗?