Scala: аргументы типа не соответствуют чертам границ параметров типа Subtractable

Question

Я новичок в scala и пытаюсь написать функцию, которая возвращает карту со всеми индексами для каждой буквы в данной строке. Мой код:

def group(string: String) = {
  val map = mutable.Map[Char, ListBuffer[Int]]()
  for (i <- string.indices) {
    val ch = string(i)
    if(map.contains(ch)) map(ch) += i
    else map += (ch -> ListBuffer(i))
  }
  map
}

Когда я пытаюсь скомпилировать, у меня появляется ошибка:

Ошибка: (14, 30) аргументы типа [?, Iterable [Any] с PartialFunction [Int с Char, Any] с scala.collection.generic.Subtractable [_>: Int с Char <: AnyVal, Iterable [Any] ] с PartialFunction [Int с Char, Any] с scala.collection.generic.Subtractable [_>: Int с Char <: AnyVal, Iterable [Any] с PartialFunction [Int с Char, Any] с scala.collection.generic.Subtractable [_>: Int с Char <: AnyVal, Equals]] {def seq: Iterable [Any] с PartialFunction [Int с Char, Any]}] {def seq: Iterable [Any] с PartialFunction [Int с Char, Any] {def seq: Iterable [Any] с PartialFunction [Int with Char, Any]}}] не соответствует границам параметра типа trait Subtractable [A, + Repr <: scala.collection.generic.Subtractable [A, Repr]] val v = for (i <- string.indices) {</p>

Кажется, что-то не так со значением цикла. Итак, я добавил в последнюю строку цикла 'true' и теперь все работает нормально:

def group(string: String) = {
  val map = mutable.Map[Char, ListBuffer[Int]]()
  for (i <- string.indices) {
    val ch = string(i)
    if(map.contains(ch)) map(ch) += i
    else map += (ch -> ListBuffer(i))
    true
  }
  map
}

Что не так в моем коде, и как я могу это исправить? Версия Scala: 2.12.6

Answer 1

Как вы можете видеть в https://docs.scala -lang.org / tutorials / FAQ / yield.html

Scala «для понимания» - это синтаксический сахар для объединения нескольких операций с foreach, map, flatMap, filter или withFilter. Scala фактически переводит for-expression в вызовы этих методов, поэтому любой класс, предоставляющий их, или их подмножество, можно использовать для понимания.

Итак, ваша for петля

for (i <- string.indices) {
  val ch = string(i)
  if(map.contains(ch)) map(ch) += i
  else map += (ch -> ListBuffer(i))
}

эквивалентно

string.indices.foreach(i => {
  val ch = string(i)
  if(map.contains(ch)) map(ch) += i
  else map += (ch -> ListBuffer(i))
})

foreach интерфейс метода

def foreach[U](f: A => U): Unit

map(ch) += i возвращает ListBuffer, но map += (ch -> ListBuffer(i)) возвращает Map. И когда компилятор пытается определить U в foreach аргументе f: Int => U, он получает что-то между ListBuffer и Map и не компилирует его.

Кроме того, компилятор не проверяет тип результата выражения if-else, если вы его где-то не используете.

Вы можете исправить свой код, переписав его следующим образом

def group(string: String) = {
    val map = mutable.Map[Char, ListBuffer[Int]]()

    def update(i: Int): Unit = {
        val ch = string(i)
        if(map.contains(ch)) map(ch) += i
        else map += (ch -> ListBuffer(i))
    }

    for (i <- string.indices) update(i)
    map
}

Но лучше использовать стандартные методы

def group(string: String) = string.toCharArray.zipWithIndex.groupBy(_._1).mapValues(_.map(_._2))

Answer 2

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

def group(string: String): mutable.Map[Char, List[Int]] = {

    val map = mutable.Map.empty[Char, List[Int]]

    string.zipWithIndex.foreach { case (char: Char, index: Int) =>
        map += (char -> (index :: map.get(char).getOrElse(List())))
    }

    map
}

Answer 3

Причина, по которой ваш метод не компилируется, заключается в том, что вы возвращаете совершенно другие типы из выражения if-else. Один возвращает ListBuffer[Int], а другой возвращает Map[Char, ListBuffer[Int]].

То, что вы хотите:

def group(string: String): mutable.Map[Char, ListBuffer[Int]] = {
  val map = mutable.Map[Char, ListBuffer[Int]]()
  for (i <- string.indices) {
    val ch = string(i)
    if (map.contains(ch)) {
      map(ch) += i
      map
    } else map += (ch -> ListBuffer(i))
  }
  map
}

Дополнительный подход без изменяемого Map или ListBuffer может быть:

def group(s: String): Map[Char, Int] = {
  s.split("\\W+")
   .flatten
   .zipWithIndex
   .groupBy { case (char, _) => char }
   .mapValues { arr => arr.map(_._2) }
}