计数频率值

Question

假设我有一个像

data T = A | B | C deriving (Enum) 

一个枚举和输入枚举值的列表：

[B, C, C, A, C, A, C] 

我正在寻找的是，鉴于这样的功能输入，返回每个元素在输入中出现的频率。输出的简单形式是频率列表（在这种情况下为[2, 1, 4]），但这不是要求。我目前的做法是这样的：

countEnum :: Enum a => [a] -> [a] -> [Word] 

countEnum elems = 
    let f x = map (fromIntegral . fromEnum . (fromEnum x ==)) [0 .. length elems - 1] 
    in foldr (zipWith (+)) (replicate (length elems) 0) . map f 

这工作，但我看到至少有两个问题：

1. 它使用length功能。
2. 它要求调用者在第一个参数中指定所有可能的值。

有没有办法改善这种情况？

Answer 1

通常比排序列表有点快正在使用Map,

enumFreq :: Enum a => [a] -> Map Int Word 
enumFreq = foldl' (\mp e -> Map.insertWith' (+) (fromEnum e) 1 mp) Map.empty 

，你可以得到

• 频率仅为每Map.elems $ enumFreq list
• 的对(value,frequency)每[(toEnum i, f) | (i,f) <- Map.assocs $ enumFreq list]

如果你的类型本身就是Ord，你可以跳过fromEnum和toEnum。

如果你有Ix和Bounded实例和类型没有太多的元素，

import Data.Array.Unboxed 

enumFreq :: (Ix a, Bounded a) => [a] -> UArray a Word 
enumFreq = accumArray (+) 0 (minBound,maxBound) . (`zip` repeat 1) 

具有更好的渐进性，使用较少的内存和更快已经是相当短名单。 （但是，这取决于类型的元素存在于名单的比例很高。）

Answer 2

也许这样？

import Control.Arrow ((&&&)) 
import Data.Function (on) 
import Data.List (groupBy, sortBy) 

data T = A | B | C deriving Enum 

countEnum :: Enum a => [a] -> [Int] 
countEnum = map length . groupBy ((==) `on` snd) . sortBy (compare `on` snd) . map (id &&& fromEnum) 

例如：

> countEnum [B, C, C, A, C, A, C] 
[2,1,4] 

如果你可以定义一个Bounded实例T则有可能数为零事件：

countEnum' :: (Bounded a, Enum a) => [a] -> [Int] 
countEnum' = map pred . countEnum . (++ enumFromTo minBound maxBound) 

> countEnum' [C, C, A, C, A, C] 
[2,0,4] 

Answer 3

如果你有Ord，您可以通过使用

import Control.List 
import Control.Arrow 

map (head &&& length) $ group $ sort elems