aboutsummaryrefslogtreecommitdiffstats
path: root/day21/part2.clj
blob: 86362f494c4719497e2a610a70e80dc7ce65a529 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
(def rolls {3 1 4 3 5 6 6 7 7 6 8 3 9 1})

(defn advance-pos [pos roll] (let [n (+ pos roll)] (if (> n 10) (- n 10) n)))

(defn add-roll [state roll p1-turn]
  (let [pos   (if p1-turn :pos1 :pos2)
        score (if p1-turn :score1 :score2)
        new-pos (advance-pos (state pos) roll)]
    (-> state
        (assoc pos new-pos)
        (update score + new-pos))))

(defonce wins (atom [0 0]))

(loop [turn 0 states {{:pos1 10 :score1 0 :pos2 1 :score2 0} 1}]
  (if (empty? states)
    (println (apply max @wins))
    (recur
      (if (= 0 turn) 1 0)
      (reduce
        #(let [kvp (first %2)]
           (if (< ((key kvp) (if (= 0 turn) :score1 :score2)) 21)
             (if (contains? %1 (key kvp))
               (update %1 (key kvp) + (val kvp))
               (conj %1 kvp))
             (do (swap! wins update turn +' (val kvp)) %1)))
        {}
        (for [s states r rolls]
          {(add-roll (first s) (key r) (= 0 turn)) (*' (second s) (val r))})))))