在 Clojure 中优化数组突变

Question

阅读了几篇关于该主题的博文后，我发现在 Clojure 中对数组进行变异是这样的：

(defn m [xs ys] 
  (dotimes [i (count xs)] 
    (aset #^ints ys (int i) 
    (int (* (int 3) (int (aget #^ints xs (int i))))))))

(def xs (into-array Integer/TYPE (range 1000000))) 和 (def ys (into-array Integer/TYPE (range 1000000))) 的位置

根据 Criterium 平均需要 14 毫秒，而 Java 也是如此，

public static int[] m(int[] x, int[] y)
{
  for(int i=0; i<x.length; i++)
    y[i] = 3*x[i];
  return y;
}

平均需要 800us。 **

我是否正在尽我所能让事情进展得更快？还有什么我可以沿着优化路径走下去的吗？

** 我使用 Criterium 和 (report-result (bench (m xs ys )) :verbose) 和 (report-result (bench (. Test m xs ys)) :verbose) 对这些进行计时

Answer 1

在 Clojure 1.3 上试试这个：

(set! *unchecked-math* true)

(defn m [^longs xs ^longs ys]
  (dotimes [i (count xs)]
    (aset ys i
      (* 3 (aget xs i)))))

Answer 2

如果您想要速度，您需要进入原始世界，在完成之前不要离开它。从盒装的Integer i 开始，然后在每个使用站点将其转换为原语是没有用的。也许你可以让dotimes 产生整数（类型提示i 的声明），但不确定。我所知道的是一个 loop-recur 构造，带有循环变量的原始初始化器：(loop [i (int 0)] ... (recur (unchecked-inc i))。此外，在您的示例中，您有(int 3)。您需要提前let，以免在每次迭代中重复拆箱。

顺便说一句，您可以使用 (int-array (range 1000000)) 创建初始化数组，而只使用 (int-array 1000000) 创建空数组。

更新

从 Clojure 1.3 开始，由于对原语的支持得到了增强，我上面写的大部分内容不再适用。 dotimes 已经使用了原始算法，所以你只需要编写来获得完整的性能是

(dotimes [i (alength ^ints xs)] 
  (aset ^ints ys i (unchecked-multiply (aget ^ints xs i) 3)

基本上，不需要int 构造函数，而是使用unchecked-multiply。