高階函數(shù)

傳入或者返回函數(shù)的函數(shù)

1.常見(jiàn)高階函數(shù)

forEach

   val list = listOf(1,2,3)
   list.forEach(::println)

map

    val list = listOf(1,2,3)
    val newList = list.map {
        it*2+1
    }
    newList.forEach(::println)

flatMap

    val mlist = listOf(
            1..20,
            2..5
    )
    val flatList = mlist.flatMap{
            it.map {
                "No. $it"
            }
    }
    flatList.forEach(::println)

reduce

累積數(shù)據(jù)處理

  val mlist = listOf(
        1..20,
        2..5
   )
   val flatList = mlist.flatMap{ it }
   println(flatList.reduce{ acc,i -> acc + i })

fold

添加初始值

  (0..6).map(::factorial).fold(5){acc, i ->  acc+i}

  //n的階乘
  fun factorial(n:Int):Int{
      if (n==0) return 1
      return (1..n).reduce{acc, i -> acc*i }
  }

filter

  println((0..6).map(::factorial).filter { it%2 == 1 })

takeWhile

當(dāng)遇到不符合條件的退出

  println((0..6).map(::factorial).takeWhile { it%2 == 1 })

let/apply

  data class Person(val name:String, val age:Int){
      fun work(){
          println("$name is working!!")
      }
  }
  fun main(args: Array<String>) {
      findPerson()?.let { (name, age)->
          println(name)
          println(age)
      }
      findPerson()?.apply { 
          work()
          println(age)
      }
  }
  fun findPerson():Person?{
      return null
  }

with

    val br = BufferedReader(FileReader("hello.txt"))
    with(br){
        var line:String?
        while (true){
            line = readLine()?:break
            println(line)
        }
        close()
    }

2.尾遞歸優(yōu)化

• 遞歸的一種特殊形式
• 調(diào)用自身后物其他操作
• tailrec關(guān)鍵字提示編譯器尾遞歸優(yōu)化
• 尾遞歸與迭代的關(guān)系

//尾遞歸
tailrec fun findListNode(head:ListNode?, value:Int):ListNode?{
    head?:return null
    if (head.value == value) return head
    return findListNode(head.next, value)
}
//非尾遞歸
fun factorial(n:Long):Long{
    return n* factorial(n-1)
}

3.閉包

• 函數(shù)運(yùn)行的環(huán)境
• 持有函數(shù)運(yùn)行狀態(tài)
• 函數(shù)內(nèi)部可以定義函數(shù)，也可以定義類

//fun add(x:Int) = fun(y:Int) = x+y

fun add(x:Int):(Int)->Int{
    return fun(y:Int):Int{
        return x+y
    }
}

4.函數(shù)復(fù)合

val add5 = {i:Int -> i+5}
val multiplyBy2 = {i:Int -> i*2}

infix fun <P1, P2, R> Function1<P1, P2>.andThen(function:Function1<P2,R>):Function1<P1,R>{
    return fun(p1:P1):R{
        return function.invoke(this.invoke(p1))
    }
}

fun main(args: Array<String>) {
    println(multiplyBy2(add5(8)))

    val infixFun = add5 andThen multiplyBy2
    println(infixFun(8))
}

5.顆粒化

• 簡(jiǎn)單說(shuō)就是多元函數(shù)變換成一元函數(shù)調(diào)用鏈

fun log(tag:String, target:OutputStream, message:Any?){
    target.write("[$tag]$message\n".toByteArray())
}

//顆粒化
fun log(tag:String)
    = fun(target:OutputStream)
    = fun(message:Any?)
    = target.write("[$tag]$message\n".toByteArray())


fun main(args: Array<String>) {
    log("benny", System.out, "helloWorld")
    log("benny")(System.out)("HelloWorld Again!")
}

6.偏函數(shù)

• 傳入部分參數(shù)得到的新函數(shù)

fun log(tag:String, target:OutputStream, message:Any?){
    target.write("[$tag]$message\n".toByteArray())
}

fun main(args: Array<String>) {
    log("benny", System.out, "helloWorld")
    
    val consoleWithTag = (::log.curried())("benny")(System.out)
    consoleWithTag("Hello World Again Again!")
    consoleWithTag("Hello World Again Again Again!")
}

fun <P1,P2,P3,R> Function3<P1,P2,P3,R>.curried()
= fun(p1:P1) = fun(p2:P2) = fun(p3:P3) = this(p1, p2, p3)