Карта одной записи карты - PullRequest
Новая
       4

Карта одной записи карты

5 голосов
/ 11 января 2012

Я хочу добиться чего-то вроде следующего: 

(_ : Map[K,Int]).mapKey(k, _ + 1)

А функция mapKey применяет свой второй аргумент (Int => Int) только к значению, хранящемуся в k. Есть ли что-то внутри стандартной библиотеки? Если нет, держу пари, что-то есть в Скалазе.

Конечно, я могу написать эту функцию сам (m.updated(k,f(m(k))), и это легко сделать. Но я сталкивался с этой проблемой несколько раз, так что, может быть, она уже решена?

Для Скалаза я представляю что-то в следующем коде: 

(m: Map[A,B]).project(k: A).map(f: B => B): Map[A,B]

Ответы [ 3 ]

3 голосов
/ 13 февраля 2012

Вот один из способов: 

scala> val m = Map(2 -> 3, 5 -> 11)
m: scala.collection.immutable.Map[Int,Int] = Map(2 -> 3, 5 -> 11)

scala> m ++ (2, m.get(2).map(1 +)).sequence
res53: scala.collection.immutable.Map[Int,Int] = Map(2 -> 4, 5 -> 11)

scala> m ++ (9, m.get(9).map(1 +)).sequence
res54: scala.collection.immutable.Map[Int,Int] = Map(2 -> 3, 5 -> 11)

Это работает, потому что (A, Option[B]).sequence дает Option[(A, B)].(sequence в целом выворачивает типы наизнанку. Т.е. F[G[A]] => [G[F[A]], учитывая F : Traverse и G : Applicative.)

3 голосов
/ 11 января 2012

Вы, конечно, можете добавить 

def changeForKey[A,B](a: A, fun: B => B): Tuple2[A, B] => Tuple2[A, B] = { kv =>
  kv match {
    case (`a`, b) => (a, fun(b))
    case x => x
  }
}

val theMap = Map('a -> 1, 'b -> 2)
theMap map changeForKey('a, (_: Int) + 1)
res0: scala.collection.immutable.Map[Symbol,Int] = Map('a -> 2, 'b -> 2)

Но это обойдёт любую оптимизацию в отношении повторного использования памяти и доступа.

Я также придумал довольно многословное и неэффективное скалярное решение с использованием молнии для предложенного вами project метода: 

theMap.toStream.toZipper.flatMap(_.findZ(_._1 == 'a).flatMap(elem => elem.delete.map(_.insert((elem.focus._1, fun(elem.focus._2)))))).map(_.toStream.toMap)

или 

(for {
  z <- theMap.toStream.toZipper
  elem <- z.findZ(_._1 == 'a)
  z2 <- elem.delete
} yield z2.insert((elem.focus._1, fun(elem.focus._2)))).map(_.toStream.toMap)

Возможно, мало пользы. Я просто публикую для справки.

1 голос
/ 12 января 2012

Вы можете сделать это так, чтобы она создала новую карту на основе старой: 

class MapUtils[A, B](map: Map[A, B]) {
  def mapValueAt(a: A)(f: (B) => B) = map.get(a) match {
    case Some(b) => map + (a -> f(b))
    case None => map
  }
}

implicit def toMapUtils[A, B](map: Map[A, B]) = new MapUtils(map)

val m = Map(1 -> 1)
m.mapValueAt(1)(_ + 1)
// Map(1 -> 2)
m.mapValueAt(2)(_ + 1)
// Map(1 -> 1)
...