2

我在clojure中实现了快速功率算法:

(defn fast-pow [a n]
  (cond (zero? n) 1
    (even? n) (letfn [(square [x] (*' x x))]
                (square (fast-pow a (/ n 2))))
    :else (*' a (fast-pow a (dec' n)))))

现在我想玩类型提示和 java 互操作。我想这样做是因为更好地理解 clojure 中的所有这些“java 东西”。看起来他们很容易,但实际上有很多隐藏的障碍。所以,我写道:

(defn ^java.math.BigInteger fast-pow 
  ([^java.lang.Long a ^java.lang.Long n]
   (fast-pow (java.math.BigInteger/valueOf a) (java.math.BigInteger/valueOf n)))
  ([^java.math.BigInteger a ^java.math.BigInteger n]
   (cond (zero? n) java.math.BigInteger/ONE
         (even? n) (letfn [(square [x] (.multiply x x))]
                    (square (fast-pow a (.divide n (java.math.BigInteger/valueOf 2)))))
         :else (.multiply a (fast-pow a (.subtract n BigInteger/ONE))))))

当然它甚至没有编译,因为错误的arity问题。所以我用谷歌搜索了如何通过 clojure 中的类型进行调度并找到multimethods. 其实在这一点上我是很天真和兴奋的,我从来没有尝试multimethods过,所以,我写了一些类似的东西:

(derive java.math.BigInteger ::biginteger)
(derive java.lang.Long ::long)
(defmulti fast-pow (fn [a n] [(class a) (class n)]))

(defmethod fast-pow [::biginteger ::biginteger] [a n]
  (cond (zero? n) java.math.BigInteger/ONE
        (even? n) (letfn [(square [x] (.multiply x x))]
                    (square (fast-pow a (.divide n (java.math.BigInteger/valueOf 2)))))
        :else (.multiply a (fast-pow a (.subtract n BigInteger/ONE)))))
(defmethod fast-pow [::long ::long] [a n] 
  (fast-pow 
   (java.math.BigInteger/valueOf a) 
   (java.math.BigInteger/valueOf n)))

事情变得有点复杂。这很好用,但我想知道是否有一种更简洁的方法来处理类型重载之类的事情。我不明白为什么在 clojure 中我不能这样做,即使在 Java 中我可以。

PS:当然我可以“隐藏”基于java.math.BigInteger嵌套函数内部的逻辑,并从外部函数调用它fast-pow,但实际上 - 这对我来说并不有趣,我很确定,这个问题可以通过使用来解决multimethods。如果我错了或遗漏了什么 - 请向我解释.. ;)

4

1 回答 1

3

如果您想根据多个参数的类型进行调度,那么多方法是合适的工具。请记住,这已经内置到所有促进数学函数中,例如 +' 和 *'。

user> (+' 2N 3)
5N

在 clojure 中,您不能使用类型提示来完成这种类型的重载/调度(这是故意完成的)。如果您仅基于第一个参数进行调度,则应该使用协议,因为它们要快得多

于 2016-04-05T23:22:45.807 回答