これは、組み込みの Memo.memo 関数で実現できます。Memo.memo は、関数 F => K => V から Memo インスタンスを作成します。これにより、基になるトライに簡単にアクセスすることもできます。例えば:
scala> def trieMemo[A, B](trie: collection.concurrent.TrieMap[A, B]) =
Memo.memo[A, B](f => k => trie.getOrElseUpdate(k, f(k)))
trieMemo: [A, B](trie: scala.collection.concurrent.TrieMap[A,B])scalaz.Memo[A,B]
scala> val trie = collection.concurrent.TrieMap[Int, Int]()
trie: scala.collection.concurrent.TrieMap[Int,Int] = TrieMap()
scala> val f = trieMemo(trie)(n => n * n)
f: Int => Int = <function1>
scala> f(5)
res0: Int = 25
scala> f(10)
res1: Int = 100
scala> trie
res2: scala.collection.concurrent.TrieMap[Int,Int] = TrieMap(5 -> 25, 10 -> 100)