排序haskell中的列表列表

时间:2010-02-21 23:25:21

标签: sorting haskell list set

我完全不知道如何编写一个函数,给定一个集合列表返回按大小拆分为子列表的集合(以及按照它们包含的集合大小排序的子列表)。

示例输入

*Main> allSets
[[1,2],[8],[1,4,7,8],[5],[1,4],[1],[2,3],[1,2,5,8],[3,4,6,7],[1,2,3,4],[4],[5,6,7,8],[3,4],[3],[2,3,5,6],[7],[6],[2]]

示例输出

*Main> collectByLength allSets
[[[2],[6],[7],[3],[4],[1],[5],[8]],[[3,4],[2,3],[1,4],[1,2]],[[2,3,5,6],[5,6,7,8],[1,2,3,4],[3,4,6,7],[1,2,5,8],[1,4,7,8]]]

基本上,它应该将所有相同大小的集合分组到它们自己的集合中,然后它将下一个最大大小的集合分组。

2 个答案:

答案 0 :(得分:10)

你正在使用“sets”这个词,但你的代码实际上使用了列表......所以,这是一个基于列表的解决方案(如果你想切换到实际的设置,很容易适应):

import Data.List (sortBy)
import Data.Function (on)

groupBy ((==) `on` length) $ sortBy (compare `on` length) [[0],[1,2],[3]]
-- => [[[0],[3]],[[1,2]]]

答案 1 :(得分:3)

导入Data.Function,Data.Ord和List后,您可以这样写:

sortBy (comparing length) $ groupBy ((==) `on` length) $ sortBy (comparing length) theList