Scala数组操作

数组要点

若长度固定则使用Array，若长度可能有变化则使用ArrayBuffer；提供初始值时不要使用new；用()来访问元素；用for(elem<-arr)来遍历元素；用for(elem<-arr if...)...yield...来将原数组转型为新数组；Scala数组和Java数组可以互操作，用ArrayBuffer，使用scala.collection.JavaConversions中的转换函数。

例子：

import scala.collection.mutable.ArrayBuffer;
object HelloWorld {
  def main(args: Array[String]) {
    //实现一个可变的数组（数组缓冲），以下两种方式等价
    val mutableArr = ArrayBuffer[Int]();
    val mutableArr2 = new ArrayBuffer[Int];
    mutableArr += 1; //用+=在尾端添加元素
    mutableArr +=(1, 2, 3, 4, 5) //在尾端添加多个元素，以括号括起来
    mutableArr ++= Array(8, 9, 10) //你可以使用++=操作符追加任何集合
    mutableArr.trimEnd(7) //移除最后7个元素
    println("length=" + mutableArr.length)
    println("size=" + mutableArr.size)
    print("元素=");
    for (elem <- mutableArr) {
      print(elem + " , ")
    }
  }
}

如果你需要在Array和ArrayBuffer之间转换，那么分别调用toArray()和toBuffer()方法即可。

数组转换

如下使用for推导式可以产生一个全新的数组

object HelloWorld {
  def main(args: Array[String]) {
    val a = Array(2, 3, 5, 6);
    val result = for (elem <- a) yield elem * 2
    result.copyToArray(a)
    for (elem <- a) {
      print(elem + ",") //output:4,6,10,12,
    }
  }
}

for(...)yield循环创建了一个类型与原始集合相同的新集合，如果你从数组出发，那么你得到的是另一个数组，如果你从数组缓冲出发，那么在for(...)yield之后得到的也是一个数组缓冲。可以使用如下方式来对满足一定条件的元素进行操作：

 //对元素值是偶数的进行翻倍操作，并舍弃掉值是奇数的元素
    val b = a.filter(_ % 2 == 0).map(_ * 2);
    for (elem <- b) {
      print(elem + ",")
    }

再看一个例子：

import scala.collection.mutable.ArrayBuffer
object HelloWorld {
  def main(args: Array[String]) {
    //移除除第一个负数之外的所有负数
    val a = ArrayBuffer(1, -2, 3, 5, -4, -5)
    var first = true
    //首先收集需要保留的下标，返回的结果是一个vector集合
    val indexs = for (i <- 0 until a.length if first || a(i) >= 0) yield {
      if (a(i) < 0) first = false; i
    }
    print(indexs) //output：Vector(0, 1, 2, 3)
    //然后将元素移动到该去的位置，并截断尾端
    for (j <- 0 until indexs.length) {
      a(j) = a(indexs(j))
    }
    println()
    a.trimEnd(a.length - indexs.length)
    print(a)
  }
}

多维数组

object HelloWorld {
  def main(args: Array[String]) {
    val matrix = Array.ofDim[Double](3, 4) //构造一个三行四列的二维数组
    matrix(0)(0) = 56 //访问第一个元素
    val triangle = new Array[Array[Int]](10) //创建一个十行的数组，每一行都是一个Int型的Array
    for (i <- 0 until (triangle.length)) {
      triangle(i) = new Array[Int](i + 1) //将每一行实例化成(i+1)列的二维数组
    }
    //打印数组
    for (i <- 0 until (triangle.length)) {
      for (j <- 0 until (triangle(i).length)) {
        print(triangle(i)(j) + ",")
      }
      println()
    }
  }
}