在向量中关联多个元素的惯用方式

Question

我有一个一维向量，一个要在向量内更新的索引向量，以及一个应该与这些索引中的每一个相关联的值。

我是 Clojure 的新手，我想可能有一种更惯用的方式来编写我最终得到的例程：

(defn update-indices-with-value [v indices value]
  (loop [my-v v 
         my-indices indices
         my-value value]
    (if (empty? my-indices)
      my-v
      (recur (assoc my-v (peek my-indices) my-value)
             (pop my-indices)
             my-value)))) 

我知道 assoc 可用于更新关联集合中的多个键或索引，但我无法弄清楚将 assoc 与任意键或索引列表一起使用的语法魔法。

Answer 1

使用reduce:

(defn assoc-all
  [v ks value]
  (reduce #(assoc %1 %2 value) v ks))

例子：

(assoc-all [1 2 3 4] [0 1] 2)
;; => [2 2 3 4]

---

或者，创建键/值对并使用apply：

(defn assoc-all
  [v ks value]
  (->> (interleave ks (repeat value))
       (apply assoc v)))

如果assoc 在内部使用瞬变，这实际上可能比reduce 版本更有效。既然不是，懒惰的序列开销可能会吃光它。

所以，最后是一个临时版本：

(defn assoc-all
  [v ks value]
  (persistent!
    (reduce
      #(assoc! %1 %2 value)
      (transient v)
      ks)))