Czy korzystanie z toList jest złe?

Question

Załóżmy 2 Mapy

import qualified Data.Map as M 
sparse1, sparse2 :: M.Map Int Float 
sparse1 = M.fromList [(1,2.0),(10,3),(12,5),(100,7),(102,11)] 
sparse2 = M.fromList [(2,13.0),(11,17),(12,19),(101,23),(102,29)] 

Jak zdefiniować elegancką funkcję

combi :: M.Map Int Float -> M.Map Int Float -> Float 

takie, że combi sparse1 sparse2 zwraca 414,0 (= 5 * 19 + 11 * 29), ponieważ 12 i 102 są jedynym wspólne klucze dwóch map? Jest to elegancki (proste i wydajne) funkcja z listami od tych, byłaby ściśle uporządkowane:

combiList xs ys = cL xs ys 0 
cL [] _ acc = acc 
cL _ [] acc = acc 
cL (x@(k,r):xs) (y@(k',r'):ys) acc 
    | k < k' = cL xs  (y:ys) acc 
    | k == k' = cL xs  ys  (acc+r*r') 
    | k > k' = cL (x:xs) ys  acc 

Ale jest

combi m1 m2 = combiList (M.toList m1) (M.toList m2) 

dobrym pomysłem wiedząc wykazy nie są już używane w pozostałej części kod ? A jeśli nie, to jak wydajnie pisać kombi bez toList?

Answer 1

Korzystanie fold i intersectWith na mapach jest nieco bardziej elegancki (i prawdopodobnie szybciej):

combi :: M.Map Int Float -> M.Map Int Float -> Float 
combi x y = M.fold (+) 0 $ M.intersectionWith (*) x y 

combi sparse1 sparse2 powraca 414.0 jako pożądane.

Jeśli zależy Ci na wydajności, spróbuj użyć Data.IntMap: tutaj powinno być kilka razy szybciej niż Data.Map.