迭代Clojure向量

Question

我正在實現一個Clojure函數(gol [coll]) ，該函數接收一個大小為1和0的向量的向量，對其進行迭代以檢查每個索引的附近位置並返回一個新的板； 像康威的人生游戲

輸入：

`(gol [[0 0 0 0 0]
       [0 0 0 0 0]
       [0 1 1 1 0]
       [0 0 0 0 0]
       [0 0 0 0 0]])`

輸出：

`[[0 0 0 0 0]
  [0 0 1 0 0]
  [0 0 1 0 0]
  [0 0 1 0 0]
  [0 0 0 0 0]]`

如何迭代向量並同時更改值？

Answer 1

使用assoc-in ：

(assoc-in v [0 0] 1)

上面將左上角的值設置為1 。

要一次設置多個，可以減少assoc-in 。

(def new-values [[[0 0] 1] 
                 [[0 1] 2] 
                 [[0 2] 3]])

(reduce
  (fn [acc ele]
    (apply assoc-in acc ele))
  v
  new-values)

;;=> [[1 2 3 0 0] ...]

要從輸入到輸出，轉換將是：

[[[2 1] 0]
 [[2 3] 0]
 [[1 2] 1]
 [[3 2] 1]]