程序员

学习swift高级函数的使用

2019-03-04  本文已影响0人  墨_辰

前两天在看队列和栈的时候没有看懂最后的reduce函数觉得很遗憾,所以就学习了一下swift提供的高级函数,Map,Flatmap,filter和reduce四个函数。这四个函数配合闭包使用会很方便。
swift自动为闭包提供参数名缩写功能,可以直接通过0和1等来表示闭包中的第一个第二个参数,并且对应的参数类型会根据函数类型来进行判断。

Map

map方法作用是把数组[T]通过闭包函数把每一个数组中的元素变成U类型的值,最后组成数组[U]。flatMap 更加强大,可以传入N个处理方法,将处理后得到数据,组合到同一个数组中
定义:func map(transform: (T) -> U) -> [U]

func map<T, U>(xs: [T], f: T -> U) -> [U]
{ 
    var result: [U] = []
    for x in xs 
    {
        result.append(f(x)) 
    }
    return result 
}

实例:
将数组中每个数字都加10,获得一个新的数组:

let numberArray = [1,2,3,4,5]

//采用闭包的方式
var result = numberArray.map({($0) + 10})
print(result)

//flatMap,2个处理方法
var resultArray = numberArray.flatMap({["\($0)个","\($0 )只"]})
print(resultArray)

filte

filter就是筛选的功能,参数是一个用来判断是否筛除的筛选闭包,根据闭包函数返回的Bool值来过滤值。为True则加入到结果数组中。
定义:func filter(includeElement: (T) -> Bool) -> [T]
例子:找出数组中大于2的数字

let numberArray = [1,2,3,4,5]
var filteredArray = numberArray.filter({$0 > 2})
print(filteredArray)

reduce

reduce的作用给定一个类型为U的初始值,把数组[T]中每一个元素传入到combine的闭包函数里面,通过计算得到最终类型为U的结果值。
定义:func reduce(initial: U, combine: (U, T) -> U) -> U
例子:求数组中所有数字的平方和

let numberArray = [1,2,3,4,5]
var sum = numberArray.reduce(0, {$0 + $1 * $1})
print(sum)

实现一下拼接地址:

let array = ["中国", "江苏", "南京", "江宁区"]
let address = array.joined(separator: "/")
let add = array.reduce( "",  {$0 + "/" + $1 } )

print(add)

添加一道面试题:

题目:实现一个函数,求0~100以内(包括0和100)中为偶数且正好是其他数字平方的数字。
代码实现:

//直接写法
func evenSquareNumbers(from : Int, to : Int) -> [Int]{
    var result = [Int]()
    
    for num in from...to where num % 2 == 0 {
        if (from...to).contains(num * num) {
            result.append(num * num)
        }
    }
    
    return result
}
//使用map和filter实现
(0...10).map { $0 * $0 }.filter { $0 % 2 == 0 }
上一篇 下一篇

猜你喜欢

热点阅读