【发布时间】:2020-09-08 20:49:02
【问题描述】:
containers 包中的Data.Map.Lazy 模块中有Map 类型(关联数组):
data Map k a = ...
我可以从这种类型派生Set:
import qualified Data.Map.Lazy as Map
newtype Set a = Set (Map.Map a ())
但同一包中的Data.Set 模块没有。是因为性能问题吗?
此外,我可以通过类似的方法推导出 C++ 的 std::multiset 和 std::multimap 等效项:
type Occur = Int
newtype MultiSet a = MultiSet (Map.Map a Occur)
newtype MultiMap k a = MultiMap (Map.Map k (MultiSet a))
因为containers 包没有提供这样的类型,我实际上是用我自己的实现来实现这些类型的。
优点是很容易为这些类型实现实用程序,例如 C++ 的 size、insert 和 erase 的(纯功能)等效项。例如:
instance Foldable MultiSet where
foldMap f (MultiSet xs) = Map.foldMapWithKey (\x o -> stimes o (f x)) xs
toList = toAscList
null (MultiSet xs) = null xs
length = size
size :: MultiSet a -> Occur
size (MultiSet xs) = sum xs
insert :: Ord a => a -> MultiSet a -> MultiSet a
insert x = insertMany x 1
insertMany :: Ord a => a -> Occur -> MultiSet a -> MultiSet a
insertMany x n (MultiSet xs) = MultiSet (Map.alter (\maybeo -> case maybeo of
Nothing -> Just n
Just m -> maybeClampNeg (m + n)
) x xs)
clampNeg :: Occur -> Occur
clampNeg n = if n < 0 then 0 else n
maybeClampNeg :: Occur -> Maybe Occur
maybeClampNeg n = case clampNeg n of
0 -> Nothing
n' -> Just n'
delete :: Ord a => a -> MultiSet a -> MultiSet a
delete x = deleteMany x 1
deleteMany :: Ord a => a -> Occur -> MultiSet a -> MultiSet a
deleteMany x n (MultiSet xs) = MultiSet (Map.update (maybeClampNeg . subtract n) x xs)
这些实现会不会出现性能问题?
【问题讨论】:
标签: dictionary haskell set multimap multiset