如何根据第一个值生成元组列表的笛卡尔连接？ [复制]

Question

我有一个(Int,String) 类型的元组列表，我想生成一个[String] 类型的列表，其中每个字符串都是具有相同Int 值的元组的第二个元素的笛卡尔连接排列。例如：

输入：[(1,"abc"),(1,"def"),(2,"ghi"),(2,"kl")]

输出：["ad", "ae", "af", "bd", "be", "bf", "cd", "ce", "cf", "gk", "gl", "hk", "hl", "ik", "il"]

我试过这个，但我找不到一种方法来用相同的 Int 值置换元组：

possible_keys :: [(Int,String)] -> [String]
possible_keys subkeys = [ key | keysize <- keysizes, key<-keys]
  where keysizes = map (fst) subkeys
        keys = sequence (map (snd) subkeys)

有什么提示吗？

Answer 1

您可以使用groupBy 将元组组分开：

import Data.Function (on)
import Data.Ord (comparing)
import Data.List

possible_keys :: Ord k => [(k, [c])] -> [[c]]
possible_keys = concat . fmap sequenceA . fmap (fmap snd)
    . groupBy ((==) `on` fst) . sortBy (comparing fst)

（我假设输入列表不一定是排序的。我只按键排序以使其侵入性最小。comparing fst 与compare `on` fst 相同；不幸的是没有类似的@ 987654328@ 还在标准库中。sequenceA 与sequence、only more general 相同。）

这可以缩小一点。第二函子定律允许我们组合fmap的连续使用：

possible_keys :: Ord k => [(k, [c])] -> [[c]]
possible_keys = concat . fmap (sequenceA . fmap snd)
    . groupBy ((==) `on` fst) . sortBy (comparing fst)

fmap 后跟sequenceA 是traverse：

possible_keys :: Ord k => [(k, [c])] -> [[c]]
possible_keys = concat . fmap (traverse snd)
    . groupBy ((==) `on` fst) . sortBy (comparing fst)

最后，fmap 后跟concat 在列表中是concatMap（或(=<<)，但这在这里有点太神秘了）：

possible_keys :: Ord k => [(k, [c])] -> [[c]]
possible_keys = concatMap (traverse snd)
    . groupBy ((==) `on` fst) . sortBy (comparing fst)

请注意，这将为只有一个元组的键生成长度为 1 的字符串 -- sequenceA ["abc"] 是 ["a","b","c"]。如果您不希望这样，您可以在 groupBy 之后立即使用 filter 元组组来摆脱只有一个元组的那些。