柯里化函数

Posted 2021-01-08 liveoutfun

柯里化函数

 

什么是柯里化？

在计算机科学中，柯里化（Currying）是把接受多个参数的函数变换成接受一个单一参数（最初函数的第一个参数）的函数，并且返回接受余下的参数且返回结果的新函数的技术。这个技术由 Christopher Strachey 以逻辑学家 Haskell Curry 命名的，尽管它是 Moses Schnfinkel 和 Gottlob Frege 发明的。

function curry(fn) {
  function _c(restNum, argsList) {
    return restNum === 0 ?
      fn.apply(null, argsList) :
      function(x) {
        return _c(restNum - 1, argsList.concat(x));
      };
  }
  return _c(fn.length, []); // 递归开始
}

function add(a,b){ return function(c){ return a+b+c; } } curry(add)(1)(2)(3);

需求分析

我们需要一个 curry 函数，它接受一个待柯里化的函数为参数，返回一个用于接收一个参数的函数，接收到的参数放到一个列表中，当参数数量足够时，执行原函数并返回结果。

实现方式

简单思考可以知道，柯里化部分配置函数的步骤数等于 fn 的参数个数，也就是说有两个参数的 plus 函数需要分两步来部分配置。函数的参数个数可以通过fn.length获取。

总的想法就是每传一次参，就把该参数放入一个参数列表 argsList 中，如果已经没有要传的参数了，那么就调用fn.apply(null, argsList)将原函数执行。要实现这点，我们就需要一个内部的判断函数 _c(restNum, argsList)，函数接受两个参数，一个是剩余参数个数 restNum，另一个是已获取的参数的列表 argsList；_c 的功能就是判断是否还有未传入的参数，当 restNum 为零时，就是时候通过fn.apply(null, argsList)执行原函数并返回结果了。如果还有参数需要传递的话，也就是说 restNum 不为零时，就需要返回一个单参数函数

function(x) {
  return _c(restNum - 1, argsList.concat(x));
}

来继续接收参数。这里形成了一个尾递归，函数接受了一个参数后，剩余需要参数数量 restNum 减一，并将新参数 x 加入 argsList 后传入 _c 进行递归调用。结果就是，当参数数量不足时，返回负责接收新参数的单参数函数，当参数够了时，就调用原函数并返回。