《Haskell趣学指南》笔记之高阶函数

1,385 阅读3分钟

系列文章


一个函数满足以下任一条件即为高阶函数

  1. 接受函数作为参数
  2. 将函数作为返回值

《计算的本质》讲得更好一些,我感觉这本书这里讲得不通透。

柯里化

让函数只接受一个参数就够了。 如果想接受两个参数,就先接受一个,返回一个接受另一个参数的函数即可。

multThree :: Int -> Int -> Int -> Int -- 等价于 Int -> (Int -> (Int -> Int))
multThree x y z = x * y * z 

截断(section)

divideByTen :: (Floating a) => a -> a 
divideByTen = (/10) -- 这就是截断
isUpperAlphanum :: Char -> Bool
isUpperAlphanum = (`elem` ['A'..' Z']) -- 这就是截断

一个需要注意的问题是 (-4) 不能表示截断,必须换成 (`subtract` 4)

函数作参数

applyTwice :: (a -> a) -> a -> a -- 注意这里的括号不能省略
applyTwice f x = f (f x)

注意这里的括号不能省略,因为 -> 默认是右结合。

技巧:在编写函数(尤其是高阶函数)时如果拿不准函数的类型,可以先不写函数的类型声明,定义完毕后再利用 :t 查看 Haskell 推导出的结果。

工具箱

ghci> map (++ "!") ["BIFF"" BANG"" POW"]
["BIFF!"," BANG!"," POW!"]
ghci> filter even [1.. 10] 
[2, 4, 6, 8, 10]
ghci> let listOfFuns = map (*) [0..] 
ghci> (listOfFuns !! 4) 5 
20

lambda

lambda 就是一次性的匿名函数。写法是\param1 param2 -> returnValue,必要的时候可以用圆括号括起来。

如果参数是二元组,应该怎么声明 lambda:

ghci> map (\(a, b) -> a + b) [(1, 2),( 3, 5)] -- 注意这里的 (a,b) 是模式匹配,不是两个参数
[3, 8]

以下例子说明 Haskell 的函数是默认柯里化的

addThree :: Int -> Int -> Int -> Int 
addThree x y z = x + y + z 

addThree' :: Int -> Int -> Int -> Int 
addThree' = \x -> \y -> \z -> x + y + z

-- 上面俩函数等价

flip1 :: (a -> b -> c) -> b -> a -> c 
flip1 f y x = f x y

flip2 :: (a -> b -> c) -> b -> a -> c 
flip2 f = \y x -> f x y

-- 上面俩函数等价

fold 折叠(类似 reduce)

foldl (\acc x -> acc + x) init list
foldr (\x acc -> x : acc) init list
foldl1 step list -- 第一个元素就是 init
foldr1 step list -- 第一个元素就是 init

另一种方式理解折叠:折叠就是对列表中的所有元素连续地应用某个函数。如:

foldr f 0 [1,2,3,4] 
-- 其实就是
f 1 (f 2 (f 3 (f 4 0)))

foldl g 0 [1,2,3,4]
-- 其实就是
g (g (g (g 0 1) 2) 3) 4

当二元函数 f 不总是需要对第二个参数求值时,即可通过 foldr 对无限列表做处理。fordl 则不行。

scan 与 fold 类似,区别在于 scan 会把每一次 step 的返回值记录在一个列表里。

$ 函数应用符号

前面的是函数, 后面的是函数的参数,类似 JS 的 .call。

sum $ filter (> 10) $ map (*2) [2.. 10] -- 得到 80
sum (filter (> 10) (map (*2) [2.. 10])) -- 如果不用 $ 就要加很多括号

ghci> map ($ 3) [(4+),( 10*),(^ 2), sqrt]
[7. 0, 30. 0, 9. 0, 1. 7320508075688772]

函数组合 composition

数学定义:(f·g)(x) = f(g(x)),它的目的是方便生成新函数。

map (negate . sum . tail) [[1.. 5],[ 3.. 6],[ 1.. 7]] 
-- [-14,- 15,- 27] 

如果要组合的函数有多个参数,就只能化为只有一个参数的函数(柯里化)再组合。

sum . replicate 5 $ max 6.7 8.9 --44.5
--等价于
sum . (replicate 5) $ max 6.7 8.9 -- 44.5

技巧:如果你打算用组合来省去大量括号,可以先找出最里面的函数和参数,写下来,在前面加一个 ,然后把其他部分组合起来写在 前面,如

replicate 2 (product (map (*3) (zipWith max [1, 2] [4, 5])))
-- 改写为
replicate 2 . product . map (*3) $ zipWith max [1, 2] [4, 5]

point-free 风格

如果你发现函数定义里出现

fn x = fn2 p1 x

那么你可以直接简化成

fn = fn2 p1

这一点在《计算的本质》也讲到了。

这样写的好处是,简洁。