Scala讲座:函数式编程处理树结构数据

开发 后端
本文继续Scala讲座第七篇的第四部分内容,本部分提供了函数式编程的比较高阶一点的案例。

在学习完函数式编程的思考方法之后,尝试一下更高级的例子吧。这次考虑一下处理类似于XML的树结构数据的程序。既不使用循环也不使用变量如何来描述复杂的处理呢?

先出一个处理XML数据的题目。例如有如下的XML数据,有目录和文件,目录下有目录和文件两种元素。

  1. < xml> 
  2. < dir name="com"> 
  3. < dir name="mamezou"> 
  4. < file name="aaa.txt">< /file> 
  5. < file name="bbb.txt">< /file> 
  6. < /dir> 
  7. < file name="ccc.txt">< /file> 
  8. < /dir> 
  9. < file name="ddd.txt">< /file> 
  10. < /xml> 

题目的内容是从中取出文件的部分,并打印出文件名。程序的执行结果因该如下:

  1. file:aaa.txt  
  2. file:bbb.txt  
  3. file:ccc.txt  
  4. file:ddd.txt 

好,会变成怎样的程序呢?另外,Scala有非常强大的XML处理功能,以上的功能实际上只要一两行程序就可以完成了。但是这次为了说明函数式编程,特地不使用哪些功能,而使用简单功能来从头开始编码。

Scala中XML语句可以作为语言文本(Literal)像数字和字符串一样被处理。像下面这样

  1. scala> val xml = < xml> 
  2. < dir name="com"> 
  3. < dir name="mamezou"> 
  4. < file name="aaa.txt">< /file> 
  5. < file name="bbb.txt">< /file> 
  6. < /dir> 
  7. < file name="ccc.txt">< /file> 
  8. < /dir> 
  9. < file name="ddd.txt">< /file> 
  10. < /xml> 
  11. xml: scala.xml.Elem =  
  12. < xml> 
  13. < dir name="com"> 
  14. :(以下略) 

没有双引号,一开始就写XML文本,然后将其赋值给变量(这里是xml)。他的类型是scala.xml.Elem,父类型为scala.xml.Node,表示XML的标记。在这里包含在< xml>< /xml>标记对中的内容被绑定在变量xml上。该Node类型里有名为child的方法,返回该标记的所有子元素。例如,这里xml.child将返回以如下两个标记为成员的类似于ArrayBuffer的数组对象。

  1. < dir name="com"> 
  2. :  
  3. < /dir> 

  1. < file name="ddd.txt"/> 

这里可以认为ArrayBuffer是列表一样的东西。进一步调用子元素的child方法则可以得到再下一层的元素。调用。< dir name="com">标签对象的child方法将返回紧邻该标签的子元素(目录标记)。

仅使用这个方法该如何写取得文件名的程序呢?如果是面向对象方式,则可以首先定义Dir类和File类,然后定义Dir和File类的抽象父类Node,然后沿着树结构定义showFiles方法,然后递归调用该方法来取得文件名。也就是所谓的组合模式(图1)。

Scala讲座 图1:组合模式 

Scala讲座 图1:组合模式

如果放弃面向对象而考虑纯粹的命令式方法的话就会很头疼了。因为只用for语句的话,对于每一个Dir都要用一个for循环,层次一多将会将会变得很复杂,这里省略了命令式方法的实现。

接下来用函数式方法来考虑一下。函数式的情况下,因为考虑的是对于各个元素应用函数,先从***元素开始考虑应用什么函数。这个函数功能是“在某一时刻返回某一元素下的文件列表”。这样就可以想到,那元素如果是file则可直接返回包含该file的列表,如果是Dir的话则返回包含所有子文件的列表。先来看看该函数的实例。

  1. def fileFinder(node:scala.xml.Node):List[scala.xml.Node] = node.label match {  
  2. case "xml" => node.child.toList.flatMap(fileFinder)  
  3. case "dir" => node.child.toList.flatMap(fileFinder)  
  4. case "file" => List(node)  
  5. case _ => List()  

其中toList()方法为将类列表对象(ArrayBuffer)转换为列表对象。刚才用的是类似于ArrayBuffer类的对象,这里将其转换为标准列表后再操作,而node.label则返回XML标记的名称。

这里开始是正题了,除了file和无匹配处理(case _ => List())部分,xml和dir处理部分是问题的关键,也就是node.child.toList.flatMap(fileFinder)部分。如果这里关注的是Node对象,那处理过程因该是这样的,首先用child方法取出Node的所有子元素,然后用前面说明过的类似于map的函数对每一个子元素应用fileFinder方法并递归重复这一过程。那为什么这样编码之后就能得到Node下的所有file元素了呢?

那么flatMap原本的功能又是什么呢?让我们将其转换成map函数,然后看一下执行过程。将XML的结构简单化之后将如下所示

  1. < xml> ←这里  
  2. < dir> 
  3. < dir> 
  4. < file name="aaa.txt"/> 
  5. < file name="bbb.txt"/> 
  6. < /dir> 
  7. < file name="ccc.txt"/> 
  8. < /dir> 
  9. < file name="ddd.txt"/> 
  10. < /xml> 

假如现在的要素位置是xml标记,将其子元素转换成列表后对其各个项目应用函数。

  1. List(fileFinder(< dir>~< /dir>), fileFinder(< file …/>)) 

file的话保持原样,如果是dir则对其子元素应用函数。

  1. List(List(fileFinder(< dir>~< /dir>),fileFinder(< file name="ccc.txt"/>)),List(< file name="ddd.txt">)) 

接着对于***个Node元素应用函数。

  1. List(List(List(< file name="aaa.txt"/>,< file name="bbb.txt"/>), List(< file name="ccc.txt"/>)), List(< file name="ddd.txt">)) 

理解上述工作过程是比较困难的,重要的是在我的脑中考虑的并不是这样复杂的逻辑,而仅仅是实现“从一个Node元素中取出file列表”的函数的逻辑。这需要一定程度的思路切换,考虑用命令式方法来实现时实际上花了我2-3小时,而想到这个函数式方法后不到10分钟就想通了。

感觉上好像已经完成了,但是这还不够。刚才用map来假想的过程完成后,得到的是List里面还有List的一个复合结构,光这样还不能被使用。那么,flatMap函数就出场了。这个函数在Scala的机制上具有同map函数同等的重要层度,将map和flatMap说成Scala函数机制的核心都不为过分。

“flatMap “函数对每一个元素应用函数参数之后将其结果以列表形式返回,这时返回结果是列表类型是关键。接着看一下简单的例子吧

首先是map函数的例子。对于内容为“1,2,3,4,5 “的列表,应用x*2函数。

  1. scala> List(1,2,3,4,5)  
  2. res134: List[Int] = List(12345)  
  3. scala> res134.map(x => x * 2)  
  4. res135: List[Int] = List(246810)  

结果是List(2, 4, 6, 8, 10),即将每一个元素乘以2。题外话,还有一个叫做filter的函数,他返回过滤结果。

  1. scala> res134.filter(x => x != 3)  
  2. res136: List[Int] = List(1245)这里是返回3以外的元素。那么,接下来对于List(12345)应用如下函数。  
  3. x => x match {  
  4. case 3 => List(3.13.23.3)  
  5. case _ => List(x * 2)  

也就是,3以外的情况下使元素值翻倍,3的时候将元素分割为“3.1, 3.2, 3.3“。因此,表面上对于List(1,2,3,4,5)适用该函数后希望返回的是List(1, 2, 3.1, 3.2, 3.3, 4, 5),但用了map函数后实际上不是。

  1. scala> res134.map(x => x match {  
  2. case 3 => List(3.13.23.3)  
  3. case _ => x * 2 
  4. | })  
  5. res138: List[Any] = List(24, List(3.13.23.3), 810

结果中的确包含了3.1, 3.2, 3.3,但是以List中包含List为形式的。这样只完成了一半,同前面的XML处理一样现象。那么,使用一下flatMap函数吧。

  1. scala> res134.flatMap(x => x match {  
  2. case 3 => List(3.13.23.3)  
  3. case _ => List(x * 2)  
  4. | })  
  5. res139: List[AnyVal] = List(243.13.23.3810

噢!就是想要的结果。不仅包含了希望的元素,还将所有元素平摊成了一个列表。

Scala讲座 图2:组合模式flatMap函数概念图 

Scala讲座 图2:组合模式flatMap函数概念图

回到XML的例子中,正因为用flatMap函数代替了map函数,所以对于< xml>和< dir>部分来说,原本在递归调用中返回的是List,但是flatMap函数将其互相合并,摊平为单一列表了。

  1. scala> def fileFinder(node:scala.xml.Node):List[scala.xml.Node] = node.label match {  
  2. case "xml" => node.child.toList.flatMap(fileFinder)  
  3. case "dir" => node.child.toList.flatMap(fileFinder)  
  4. case "file" => List(node)  
  5. case _ => List()}  
  6. fileFinder: (scala.xml.Node)List[scala.xml.Node]  
  7. scala> fileFinder(xml).foreach(x => println("file:" + x.attribute("name").getOrElse("")))  
  8. file:aaa.txt  
  9. file:bbb.txt  
  10. file:ccc.txt  
  11. file:ddd.txt 

正如所愿的结果就一下子得到了,函数式编程真是恐怖呀!这次学的map和flatMap函数在Scala中有非常重要的意义。这可以说是函数式编程的一个高潮,理解了这个之后领悟的大门就可以说向你敞开了。这实际上还与单子(monado)这一思考方法有关,理解了map和flatMap函数之后可以说是踏出了完全掌握该思考方法的一大步。关于“单子”在本连载中还会着重说明。

【编辑推荐】

  1. 万物皆对象:介绍Scala对象
  2. Scala的泛型:***大的特性
  3. Scala的Trait:可以包含代码的接口
  4. Scala的模式匹配和条件类
  5. Scala类:复数类,无参方法,继承和覆盖
责任编辑:book05 来源: JavaEye博客
相关推荐

2009-12-11 10:44:00

Scala讲座函数 scala

2009-09-24 09:41:00

Scala讲座Scala

2009-09-27 15:29:00

Scala讲座面向对象Scala

2010-03-11 10:34:22

Scala

2009-12-11 10:42:00

Scala讲座类定义构造函数

2009-09-24 09:38:00

Scala讲座第一类对象scala

2018-11-12 09:50:56

Python函数式编程数据结构

2009-07-08 16:10:24

Scala简介面向对象函数式

2010-01-28 14:51:24

Scala后函数式

2009-12-11 10:43:00

Scala讲座操作符函数

2009-06-09 13:18:56

Scala函数式命令式

2010-07-09 14:12:00

ScalaF#C#

2010-07-07 13:11:20

ScalaF#C#

2009-12-11 10:45:00

Scala讲座类型系统功能

2013-09-09 09:41:34

2009-09-24 09:28:00

Scala讲座全局变量scala

2023-05-11 08:37:46

LinuxSpring MVC

2009-07-21 17:16:34

Scala函数式指令式

2023-12-14 15:31:43

函数式编程python编程

2022-09-22 08:19:26

WebFlux函数式编程
点赞
收藏

51CTO技术栈公众号