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

Question

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

val s = if (<some condition>) Array(...) else List(...) 
print(s.length) 

Дело в том, что с будет выведено в качестве Object, так как нет общего суперкласса для Array и List. Таким образом, вторая строка дает ошибку, так как Object не имеет метода length.

Заранее благодарен!

Update

Ба, это позор, что я не могу принять несколько ответов! Спасибо за вашу помощь, ребята! Я играл с Scala довольно долгое время, но я, кажется, все еще есть много, чтобы узнать :-)

Answer 1

Вы можете использовать тот факт, что массивы неявно преобразуются в ArrayOps, что делает их доступными в качестве стандартных коллекций. В основном все, что вам нужно сделать, это выбрать подходящий интерфейс, общий для обоих ArrayOps и List и объявить его в качестве явного типа для переменной, чтобы вызвать преобразование:

val s: collection.SeqLike[Int,_] = 
    if (condition) Array(1, 2, 3) 
    else List(1, 2, 3, 4); 
print(s.size) 

Обновление: Для многомерных массивов, вам нужно вызвать неявные преобразования внутри, а также, потому что Array[ArrayOps[X]] не присваиваемые Array[Array[X]] и наоборот:

type SL[+A] = SeqLike[A,_] 
val s1: SL[SL[Int]] = 
    if (x) Array(Array(1, 2, 3): SL[Int]) else List(List(1, 2, 3, 4)); 
print(s1.size) 

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

Если вы сами создаете массивы из фиксированного набора элементов, вы можете создавать вспомогательные функции, которые возвращают одну из двух возможных оберток (ArrayOps и WrappedArray). Тогда вам не потребуется какая-либо явная типизация:

import scala.collection.mutable._ 
import scala.reflect.ClassManifest 

// Using ArrayOps 
def arrayO[A: ClassManifest](xs: A*): ArrayOps[A] = Array(xs : _*); 
val s2 = 
    if (x) arrayO(arrayO(1, 2), arrayO(3)) else List(List(1, 2), List(3, 4)); 
println(s2.size) 

// UsingWrappedArray 
def arrayW[A: ClassManifest](xs: A*): WrappedArray[A] = Array(xs : _*); 
val s3 = 
    if (x) arrayW(arrayW(1, 2), arrayW(3)) else List(List(1, 2), List(3, 4)); 
println(s3.size) 

Если у вас есть существующие массивы, которые вы хотите, чтобы обернуть, вам нужно обернуть каждый уровень, как в

val a4 = Array(Array(1, 2, 3), Array(5, 6)); 
val s4: SL[SL[Int]] = a.map(x => x: SL[Int]) 

Answer 2

Я был под впечатлением, что черта Scala Iterable охватывает этот случай использования: http://www.scala-lang.org/api/current/index.html#scala.collection.Iterable я тоже noobish о Scala.

Answer 3

Вы можете использовать обертку для Array и использовать s в Seq[T]:

scala> val s = if (true) Array(1): Seq[Int] else List(1) 
s: Seq[Int] = WrappedArray(1) 

Для многомерных массивов можно создавать свои собственные конвертеры, как это:

implicit class ArrArrWrap[T](a: Array[Array[T]]) extends IndexedSeq[WrappedArray[T]] { 
    def length = a.length 
    def apply(idx: Int) = a(idx): WrappedArray[T] 
} 

Использование:

scala> val s = if (true) Array(Array(1)): Seq[Seq[Int]] else List(List(1)) 
s: Seq[Seq[Int]] = (WrappedArray(1)) 

Идея использования мультидименси массив в виде многомерных последовательностей имеет большой недостаток:

Он включает преобразование при доступе ко всем измерениям, кроме последнего.

s(0) // <- conversion here 

Это может привести к проблемам с производительностью. Это может быть лучше, чтобы преобразовать Array в Seq:

scala> val s = if (true) Array(Array(1)).map{ i => i: Seq[Int] }(breakOut) else List(List(1)) 
s: scala.collection.immutable.Seq[Seq[Int]] = Vector(WrappedArray(1))