Миграция кода Java TreeMap в Scala?

я переношу свою базу кода Java на чистый Scala, и я застрял на этом одном куске кода. У меня есть реализация IntervalMap т. е. структуры данных, которые позволяют вам эффективно отображать диапазоны [from,to] до values здесь set, delete и get операции O(log n) (немного отличается от IntervalTree или SegmentTree).

этот код использует Java java.util.TreeMaps и во время миграции в Scala я столкнулся с 2 большими вопросы:

  1. Scala нет mutable.TreeMap - я решил обойти его с помощью mutable.TreeSet (странно Scala имеет mutable.TreeSet а не mutable.TreeMap) для хранения ключей и хранения ценностей в дополнительном mutable.Map. Это неприятный хак, но есть ли лучший способ?

  2. следующая проблема-Scala mutable.TreeSet не имеет эквивалента java.util.TreeSet ' s ceilingKey, floorEntry, pollFirst, pollLast все O(log n) операции в Ява.

Итак, как я могу лучше всего перенести свой код в Scala? Какова наилучшая практика в таких ситуациях? Я действительно не хочу писать свои собственные реализации дерева. Есть ли более идиоматические Скала способ написания IntervalMaps, что я не знаю? Или есть какая-нибудь уважаемая библиотека? Или же Скала просто сосать вот его овеянный TreeSet и несуществующие карты дерева. Конечно, я могу просто использовать Java TreeMap в Scala, но это уродливо, и я теряю все хорошие функции коллекции Scala, и я мог бы также использовать Java.

вот мой текущий код Java:https://gist.github.com/pathikrit/5574521

4 ответов


ответ, К сожалению, просто использовать Java TreeMap класса.

Scala не имеет собственной копии всего, и это одно из самых заметных исключений. Одна из причин, по которой он совместим с Java, заключается в том, что вам не нужно заново изобретать каждое колесо.

причина, по которой Вы все еще хотите использовать Scala, заключается в том, что не каждый бит кода, который вы пишете, касается этой TreeMap. Ваш IntervalMap может быть Scala IntervalMap; вы просто используете Java TreeMap внутренне реализовать его. Или вы можете использовать неизменяемую версию в Scala, которая теперь работает достаточно хорошо для неизменяемой версии.

возможно, в 2.11 или 2.12 будет Мутабельный TreeMap; требуется, чтобы кто-то написал его, протестировал, оптимизировал и т. д. но я не думаю, что есть философские возражения против этого.


1) в чем проблема с использованием непреложный дерева внутри? Это более или менее так же эффективно, как изменяемая карта дерева, делает все в O(log n).

2) в версии Scala нет ceilingKey и floorKey, но вместо этого есть методы from и to которые делают по существу то же самое, но возвращают целое поддерево вместо отдельных записей.

полный 1:1 порт Java-кода:

import scala.collection._
import scala.collection.immutable.TreeMap

case class Segment[T](start: Int, end: Int, value: T) {
  def contains(x: Int) = (start <= x) && (x < end)
  override def toString = "[%d,%d:%s]".format(start, end, value)
}

class IntervalMap[T] {
  private var segments = new TreeMap[Int, Segment[T]]
  private def add(s: Segment[T]): Unit = segments += (s.start -> s)
  private def destroy(s: Segment[T]): Unit = segments -= s.start
  def ceiling(x: Int): Option[Segment[T]] = {
    val from = segments.from(x)
    if (from.isEmpty) None
    else Some(segments(from.firstKey))
  }
  def floor(x: Int): Option[Segment[T]] = {
    val to = segments.to(x)
    if (to.isEmpty) None
    else Some(segments(to.lastKey))
  }
  def find(x: Int): Option[Segment[T]] = {
    floor(x).filter(_ contains x).orElse(ceiling(x))
  }

  // This is replacement of `set`, used as myMap(s,e) = v
  def update(x: Int, y: Int, value: T): Unit = {
    require(x < y)
    find(x) match {
      case None => add(Segment[T](x, y, value))
      case Some(s) => {
        if (x < s.start) {
          if (y <= s.start) {
            add(Segment[T](x, y, value))
          } else if (y < s.end) {
            destroy(s)
            add(Segment[T](x, y, value))
            add(Segment[T](y, s.end, s.value))
          } else {
            destroy(s)
            add(Segment[T](x, s.end, value))
            this(s.end, y) = value
          }
        } else if (x < s.end) {
          destroy(s)
          add(Segment[T](s.start, x, s.value))
          if (y < s.end) {
            add(Segment[T](x, y, value))
            add(Segment[T](y, s.end, s.value))
          } else {
            add(Segment[T](x, s.end, value))
            this(s.end, y) = value
          }
        } else {
          throw new IllegalStateException
        }
      }
    }
  }

  def get(x: Int): Option[T] = {
    for (seg <- floor(x); if (seg contains x)) yield seg.value
  }

  def apply(x: Int) = get(x).getOrElse{
    throw new NoSuchElementException(
      "No value set at index " + x
    )
  }

  override def toString = segments.mkString("{", ",", "}")
}

// little demo
val m = new IntervalMap[String]
println(m)
m(10, 20) = "FOOOOOOOOO"
m(18, 30) = "_bar_bar_bar_"
m(5, 12) = "bazzz"
println(m)

for (x <- 1 to 42) printf("%3d -> %s\n",x,m.get(x))

результат:

{}
{5 -> [5,12:bazzz],12 -> [12,18:FOOOOOOOOO],18 -> [18,20:_bar_bar_bar_],20 -> [20,30:_bar_bar_bar_]}
  1 -> None
  2 -> None
  3 -> None
  4 -> None
  5 -> Some(bazzz)
  6 -> Some(bazzz)
  7 -> Some(bazzz)
  8 -> Some(bazzz)
  9 -> Some(bazzz)
 10 -> Some(bazzz)
 11 -> Some(bazzz)
 12 -> Some(FOOOOOOOOO)
 13 -> Some(FOOOOOOOOO)
 14 -> Some(FOOOOOOOOO)
 15 -> Some(FOOOOOOOOO)
 16 -> Some(FOOOOOOOOO)
 17 -> Some(FOOOOOOOOO)
 18 -> Some(_bar_bar_bar_)
 19 -> Some(_bar_bar_bar_)
 20 -> Some(_bar_bar_bar_)
 21 -> Some(_bar_bar_bar_)
 22 -> Some(_bar_bar_bar_)
 23 -> Some(_bar_bar_bar_)
 24 -> Some(_bar_bar_bar_)
 25 -> Some(_bar_bar_bar_)
 26 -> Some(_bar_bar_bar_)
 27 -> Some(_bar_bar_bar_)
 28 -> Some(_bar_bar_bar_)
 29 -> Some(_bar_bar_bar_)
 30 -> None
 31 -> None
 32 -> None
 33 -> None
 34 -> None
 35 -> None

на Segment класс должно быть установлено private[yourPackage], некоторая документация должна быть добавлена.


похоже, вы хотите использовать функции nice Scala collections. Я не думаю, что тебе нужно переделывать свой класс.

видели scala.collection.JavaConversions?

вы можете использовать аналогичный подход с оболочкой, а затем реализовать методы, которые вы хотите, соответственно. Возможно, вам нужно быть более творческим с тем, как вы определяете, а затем используете методы, уникальные для вашего вида карты, но это не должно быть большим делом.

надеюсь, это даст вам идею. Дайте мне знать, если вам нужно больше руководство и я могли бы помочь вам (похоже, прошло некоторое время с тех пор, как вы спросили).


Scala 2.12 имеет mutable.TreeMap наконец:https://github.com/scala/scala/pull/4504