如何在Racket中实现自定义的curry函数？

核心原理 #

固定元数的curry函数核心是分步收集参数：每次接收部分参数后，判断是否满足原函数的参数数量要求，若满足则执行原函数，否则返回新的函数继续收集剩余参数。

开发步骤 #

1. 获取原函数的参数元数
   首先要确定目标函数需要的固定参数个数，在Scheme中可以用procedure-arity（或procedure-arity-min，不同实现略有差异）来获取。比如(procedure-arity cons)会返回2，对应cons需要的两个参数。

2. 设计可变参数的入口函数
   my-curry需要支持两种调用形式：仅传入原函数（如(my-curry cons)），或传入原函数加部分参数（如(my-curry cons 1)）。所以函数本身要能接收可变数量的参数，第一个参数是目标函数，后续是已收集的参数。

3. 递归判断参数是否足够
   每次调用时计算剩余需要的参数数量：原函数元数 - 已收集的参数个数。

   • 如果剩余数量为0：直接调用原函数并传入所有已收集的参数。
   • 如果剩余数量大于0：返回一个新的匿名函数，该函数会接收新的参数，将其与已收集的参数合并后，再次触发判断逻辑。

具体实现示例（Scheme） #

(define (my-curry proc . args)
  (let* ((arity (procedure-arity proc))
         ;; 兼容部分Scheme实现返回参数范围的情况，取固定元数
         (required (if (pair? arity) (car arity) arity))
         (given (length args)))
    (cond
      ;; 参数足够，直接执行原函数
      ((= given required) (apply proc args))
      ;; 参数不足，返回新函数继续收集
      ((< given required)
       (lambda rest
         (apply my-curry proc (append args rest))))
      ;; 参数过多时抛出错误（固定元数场景下的常规处理）
      (else (error "Too many arguments passed to my-curry")))))

验证示例 #

;; 测试单参数函数
((my-curry even?) 2) ; 返回 #t

;; 分步传参
(((my-curry cons) 1) 2) ; 返回 '(1 . 2)

;; 先传部分参数再补全
((my-curry cons 1) 2) ; 返回 '(1 . 2)

注意事项 #

• 不同Scheme实现对procedure-arity的返回格式可能不同，部分会返回参数范围（比如(2 . 2)表示固定2个参数），需要做简单兼容处理。
• 该实现仅适用于固定元数的函数，若传入可变参数函数（如+）会不符合预期。

内容的提问来源于stack exchange，提问作者Exodus1452