题目要求:
(**) 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) }
时间: 2024-10-07 12:46:30