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的主要内容,如果未能解决你的问题,请参考以下文章

scala 学习之:list span 用法

scala 学习之: list.fill 用法

[LeetCode] Flatten Nested List Iterator 压平嵌套链表迭代器

341. Flatten Nested List Iterator

Scala学习之Scala快速入门

Scala学习之SCALA 集合类