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可用于更新关联集合中的多个键或索引，但我无法弄清楚使用与任意键或索引列表关联的语法魔法。

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)))

在矢量中关联多个元素的惯用法

1 个答案: