scala学习之: Flatten a nested list structure

时间:2022-10-03 03:38:58

题目要求:

(**) 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)
}