Scala入门系列（十）：函数式编程之集合操作

1. Scala的集合体系结构

Scala中的集合体系主要包括（结构跟Java相似）：

Iterable（所有集合trait的根trait）
Seq（Range、ArrayBuffer、List等）
Set（HashSet、LinkedHashSet、SortedSet等）
Map （HashMap、SortedMap、LinkedHashMap等）

Scala中的集合分为可变和不可变两类集合，分别对应scala.collection.mutable和scala.collection.immutable两个包。

2. List

List代表一个不可变的列表。

List有head和tail，head代表List的第一个元素，tail代表第一个元素之后的所有元素。

scala> val list = List(1, 2, 3, 4) 
list: List[Int] = List(1, 2, 3, 4) 
 
scala> list.head 
res33: Int = 1 
 
scala> list.tail 
res34: List[Int] = List(2, 3, 4)

案例：用递归函数给List中每个元素都加上指定的前缀并打印


// 如果List只有一个元素，那么他的tail就是Nil 

def decorator(list: List[Int], prefix: String){ 
  if (list != Nil) { 
    println(prefix + list.head) 
    decorator(list.tail, prefix) 
  } 
} 
 
scala> decorator(list, "+") 
+1 
+2 
+3 
+4

List有特殊的::操作符，可以用于将head和tail合并成一个List。

scala> list 
res37: List[Int] = List(1, 2, 3, 4) 
 
scala> 0::list 
res38: List[Int] = List(0, 1, 2, 3, 4)

该操作符在Spark源码中有体现

3. LinkedList

LinkedList代表一个可变的列表，其elem和next属性类似于List的head和tail。

案例：使用while循环将LinkedList中每个一个元素乘以二。

val list = scala.collection.mutable.LinkedList(1,2,3,4,5,6,7,8,9)

var currentList = list

var first = true

while( currentList !=Nil && currentList.next != Nil){

if(first) { currentList.elem *= 2; first = false}

currentList = currentList.next.next

if(currentList != Nil) currentList.elem *= 2

}

list: scala.collection.mutable.LinkedList[Int] = LinkedList(2, 2, 6, 4, 10, 6, 14, 8, 18)