我在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
。如果我错了或遗漏了什么 - 请向我解释.. ;)