Свести вложенные объекты в Scala - PullRequest
Новая
       19

Свести вложенные объекты в Scala

0 голосов
/ 18 мая 2018

Учитывая сложный объект, подобный следующему: 

case class Complex
(
  id: Long,
  name: String,
  nested: Seq[Complex]
)

В действии это может превратиться в нечто вроде этого: 

val stuff =
  List(
    Complex(1, "name1",
      List(
        Complex(2, "name2", List()),
        Complex(3, "name3",
          List(
            Complex(4, "name4", List())
          )
        )
      )
    )
  )

Мне нужно превратить его в плоский список Complex предметов, тянущих всех детей / внуков.

val flattened =
  List(
    Complex(1, "name1", List()),
    Complex(2, "name2", List()),
    Complex(3, "name3", List()),
    Complex(4, "name4", List()),
  )

Есть ли у вас какие-либо подсказки / идеи о том, как я могу это сделать?

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

Похоже, все они дают один и тот же список, с которого я начал.

Ответы [ 4 ]

0 голосов
/ 18 мая 2018

Мое решение в основном похоже на уже опубликованные, но позволяет избежать (не столь существенной) неэффективности помещения элемента head в одноэлементную коллекцию при выравнивании одного элемента: 

def flatten(complex: Complex): Seq[Complex] =
  complex +: complex.nested.flatMap(flatten)

вместо 

def flatten(complex: Complex): Seq[Complex] =
  Seq(complex) ++ complex.nested.flatMap(flatten)

для последующего использования следующим образом: 

stuff.view.flatMap(flatten).map(_.copy(nested = Nil))

Обратите внимание, что я также отложил замену элементов nested пустым списком (Nil), отделяя его от фактическогосглаживание, при этом избегая ненужных двойных проходов, используя view (подробнее об этом здесь ).

С этим решением можно поиграться в Scastie .

0 голосов
/ 18 мая 2018 
def flattenComplex(c: Complex): Seq[Complex] = {
  if (c.nested.isEmpty) Seq(c)
  else Seq(c.copy(nested = Nil)) ++ c.nested.flatMap(flattenComplex _)
}

, что дает: 

scala> stuff.flatMap(flattenComplex _)
res0: List[Complex] = List(Complex(1,name1,List()), Complex(2,name2,List()), Complex(3,name3,List()), Complex(4,name4,List()))
0 голосов
/ 18 мая 2018

Использование рекурсии Tail. 

def flatten(source: Seq[Complex], dest: List[Complex]): List[Complex] = source match {
  case Nil => dest
  case x :: xs => flatten(xs, flatten(x.nested, dest :+ x.copy(nested = Nil)))
}
flatten(stuff, List())
0 голосов
/ 18 мая 2018

Сложность сглаживания ввода Seq здесь заключается в том, что необходимо удалить вложенные ссылки в результирующем списке.Это может быть сделано путем копирования исходного объекта с nested = пустым списком и выравнивания всех последовательностей: 

def flatten(obj: Complex): Seq[Complex] = {
  val unnested = obj.copy(nested = List())
  Seq(unnested) ++ obj.nested.flatMap(flatten)
}

println(stuff.flatMap(flatten))

List(
  Complex(1,name1,List()),
  Complex(2,name2,List()),
  Complex(3,name3,List()),
  Complex(4,name4,List())
  )
...