拼合陣列時，一些元素不是數組

Question

如果我有

var a = Array(Array(1, 2), 3, Array(4,5,6)) 

，我想將其轉換爲

Array(1, 2, 3, 4, 5, 6) 

是什麼做的最簡單的方法？在this post 中給出了一個列表的解決方案，但它不適用於數組。

我也試過

def flatArray(a:Array[Any])= a.map(x => x match { case ar:Array[_] => ar; case _ => Array(x) }) 

但輸出ArraySeq型的，我不能看到如何將其轉換爲Array

Answer 1

def flatArray[T : ClassManifest](a:Array[Any]) = 
    a.flatMap{ 
    case ar:Array[T] => ar 
    case x: T => Array(x) 
    } 

我試着也用＃扁平化方法，但在NPE上失敗。

更新：要回答Jus12的問題：

def flatArray[T : Manifest](a:Array[Any]) = 
    a.flatMap{ 
    case ar: Array[_] if ar.getClass.getComponentType == manifest[T].erasure => ar.asInstanceOf[Array[T]]; 
    case x => Array(x.asInstanceOf[T]) 
    } 

整個球場的解決方案是不是類型安全的。原因是容納編譯器的類型推斷，推斷Array(Array(1, 2), 3, Array(4,5,6))爲Array[Any]。一個準確的類型是「一個Int或Array[Int]」的數組，但這是不可能的。什麼是是，每個元素都是Either[Int, Array[Int]]和工作與創建要麼元素的數組：

object EitherView { 
    type ||[A, B] = Either[A, B] 
    // convenience of definition functions 
    private def l[A,B](a: A): ||[A,B] = Left(a) 
    private def r[A,B](b: B): ||[A,B] = Right(b) 

    // implicit defs - stuttering-or 
    implicit def aToOr2[A,B](a: A): A || B = l(a) 
    implicit def bToOr2[A,B](b: B): A || B = r(b) 
    implicit def aToOr3[A,B,C](a: A): A || B || C = l(l(a)) 
    implicit def bToOr3[A,B,C](b: B): A || B || C = l(r(b)) 
    implicit def aToOr4[A,B,C,D](a: A): A || B || C || D = l(l(l(a))) 
    implicit def bToOr4[A,B,C,D](b: B): A || B || C || D = l(l(r(b))) 
    implicit def aToOr5[A,B,C,D,E](a: A): A || B || C || D || E = l(l(l(l(a)))) 
    implicit def bToOr5[A,B,C,D,E](b: B): A || B || C || D || E = l(l(l(r(b)))) 
    // more? ... 

} 

import EitherView._ 

type CompoundArray[T] = Array[T || Array[T]] 

object CompoundArray { 
    def apply[T](elems: (T || Array[T])*) = elems.toArray 
} 

def flatArray[T : Manifest](a:CompoundArray[T]) = { 
    a.flatMap{ 
    case Left(x) => Array(x) 
    case Right(x) => x 
    } 
} 

參見：

scala> val a = CompoundArray[Int](Array(1, 2), 3, Array(4,5,6)) 
a: Array[EitherView.||[Int,Array[Int]]] = Array(Right([I@1364b53), Left(3), Right([I@18b62e0)) 

scala> flatArray(a) 
res0: Array[Int] = Array(1, 2, 3, 4, 5, 6) 

scala> flatArray(CompoundArray[String](Array("hi"), "bye")) 
res4: Array[String] = Array(hi, bye) 

scala> flatArray(CompoundArray[String](Array("hi"), 3)) 
<console>:13: error: type mismatch; 
found : Int(3) 
required: EitherView.||[String,Array[String]] 
     flatArray(CompoundArray[String](Array("hi"), 3)) 
                ^

注：EitherView最初的想法是由@米奇Blevins：http://cleverlytitled.blogspot.com/2009/03/disjoint-bounded-views-redux.html

Answer 2

與IttayD類似，沒有ClassManifest，並且Array [Any]作爲結果。

scala> a.flatMap{ 
    |   case ar: Array[_] => ar 
    |   case x => List(x) 
    | } 
    res4: Array[Any] = Array(1, 2, 3, 4, 5, 6)