scala学习之: Flatten a nested list structure
Posted 鱼儿慢慢游~~
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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) }
以上是关于scala学习之: Flatten a nested list structure的主要内容,如果未能解决你的问题,请参考以下文章
[LeetCode] Flatten Nested List Iterator 压平嵌套链表迭代器