scala学习之： Flatten a nested list structure

题目要求：

(**) Flatten a nested list structure.
Example:
scala> flatten(List(List(1, 1), 2, List(3, List(5, 8))))
res0: List[Any] = List(1, 1, 2, 3, 5, 8)

参考：http://blog.thedigitalcatonline.com/blog/2015/04/07/99-scala-problems-07-flatten/#.WBsDFPl97IU

 

针对一个list， 将其内部嵌套的元素拿出来，形成一个新的list。

思路： 使用flatMap 函数。针对每一个元素，判断其类型，如果是list 类型的，则递归，如果是元素，则加入list。

代码如下：

def flatten[any](a:List[any]):List[any] = a flatMap{
      case li:List[_] => flatten(li)
      case h => List(h)
      }