Миграция кода Java TreeMap в Scala?
я переношу свою базу кода Java на чистый Scala, и я застрял на этом одном куске кода. У меня есть реализация IntervalMap т. е. структуры данных, которые позволяют вам эффективно отображать диапазоны [from,to]
до values
здесь set
, delete
и get
операции O(log n)
(немного отличается от IntervalTree или SegmentTree).
этот код использует Java java.util.TreeMaps
и во время миграции в Scala я столкнулся с 2 большими вопросы:
Scala нет
mutable.TreeMap
- я решил обойти его с помощьюmutable.TreeSet
(странно Scala имеетmutable.TreeSet
а неmutable.TreeMap
) для хранения ключей и хранения ценностей в дополнительномmutable.Map
. Это неприятный хак, но есть ли лучший способ?следующая проблема-Scala
mutable.TreeSet
не имеет эквивалентаjava.util.TreeSet
' sceilingKey
,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