mirror of
https://github.com/Ajetski/advent-of-code.git
synced 2025-11-27 12:32:45 -10:00
simplify. refactor in terms of pt 2 solution
This commit is contained in:
parent
57c87a2c13
commit
43c69f4e0b
@ -1,89 +1,15 @@
|
|||||||
(ns day19
|
(ns day19
|
||||||
(:require
|
(:require
|
||||||
[clojure.string :as str] ; [core :as c]
|
[clojure.string :as str]
|
||||||
input-manager
|
input-manager))
|
||||||
[orchard.pp :as pp]))
|
|
||||||
|
|
||||||
(def sample-input "r, wr, b, g, bwu, rb, gb, br
|
|
||||||
|
|
||||||
brwrr
|
|
||||||
bggr
|
|
||||||
gbbr
|
|
||||||
rrbgbr
|
|
||||||
ubwu
|
|
||||||
bwurrg
|
|
||||||
brgr
|
|
||||||
bbrgwb")
|
|
||||||
|
|
||||||
(let [[w _ & p]
|
(let [[w _ & p]
|
||||||
#_(str/split-lines sample-input)
|
|
||||||
(input-manager/get-input 2024 19)]
|
(input-manager/get-input 2024 19)]
|
||||||
(def words (into #{} (str/split w #", ")))
|
(def words (into #{} (str/split w #", ")))
|
||||||
(def puzzles p)
|
(def puzzles p)
|
||||||
(def word-sizes (->> words
|
(def word-sizes (->> words
|
||||||
(map count)
|
(map count)
|
||||||
distinct
|
distinct)))
|
||||||
sort
|
|
||||||
reverse)))
|
|
||||||
|
|
||||||
(defn p [x]
|
|
||||||
(println x)
|
|
||||||
x)
|
|
||||||
|
|
||||||
(defn next-search-index [puzzle searched valid-subsets]
|
|
||||||
(if (every? (complement true?) searched)
|
|
||||||
0
|
|
||||||
(some->> valid-subsets
|
|
||||||
(map-indexed vector)
|
|
||||||
(filter second)
|
|
||||||
(filter (fn [[idx :as el]]
|
|
||||||
(not (searched (inc idx)))))
|
|
||||||
last
|
|
||||||
first
|
|
||||||
inc)))
|
|
||||||
|
|
||||||
(defn find-valid-subsets [puzzle search-idx valid-subsets]
|
|
||||||
(let [max-idx (count puzzle)]
|
|
||||||
(->> word-sizes
|
|
||||||
(map (partial + search-idx)) ; get end pos
|
|
||||||
(filter #(<= % max-idx))
|
|
||||||
(map (partial subs puzzle search-idx)) ;get substrings
|
|
||||||
(filter words)
|
|
||||||
(map (comp (partial + search-idx) dec count))
|
|
||||||
(reduce (fn [acc idx]
|
|
||||||
(assoc acc idx true))
|
|
||||||
valid-subsets))))
|
|
||||||
|
|
||||||
(defn is-valid
|
|
||||||
([puzzle]
|
|
||||||
(is-valid puzzle
|
|
||||||
(mapv (constantly false) puzzle)
|
|
||||||
(mapv (constantly false) puzzle)
|
|
||||||
0))
|
|
||||||
([puzzle searched valid-subsets n]
|
|
||||||
(let [search-idx (next-search-index puzzle searched valid-subsets)]
|
|
||||||
(cond
|
|
||||||
(or (not search-idx)
|
|
||||||
(= search-idx -1))
|
|
||||||
false
|
|
||||||
|
|
||||||
(>= search-idx (count puzzle))
|
|
||||||
true
|
|
||||||
|
|
||||||
:else
|
|
||||||
(let [next-subsets (find-valid-subsets puzzle search-idx valid-subsets)]
|
|
||||||
(if (last next-subsets)
|
|
||||||
true
|
|
||||||
(recur puzzle
|
|
||||||
(assoc searched search-idx true)
|
|
||||||
next-subsets
|
|
||||||
(inc n))))))))
|
|
||||||
|
|
||||||
;; part 1 solution
|
|
||||||
(->> puzzles
|
|
||||||
(map is-valid)
|
|
||||||
(filter identity)
|
|
||||||
count)
|
|
||||||
|
|
||||||
(defn valid-combination-count
|
(defn valid-combination-count
|
||||||
([puzzle]
|
([puzzle]
|
||||||
@ -103,10 +29,13 @@ bbrgwb")
|
|||||||
(map (partial + pos))
|
(map (partial + pos))
|
||||||
(reduce (fn [acc idx] (update acc idx (partial + weight)))
|
(reduce (fn [acc idx] (update acc idx (partial + weight)))
|
||||||
dp))]
|
dp))]
|
||||||
(recur puzzle
|
(recur puzzle (inc pos) (get dp' pos) dp')))))
|
||||||
(inc pos)
|
|
||||||
(or (get dp' pos) 0)
|
;; part 1 solution
|
||||||
dp')))))
|
(->> puzzles
|
||||||
|
(map valid-combination-count)
|
||||||
|
(filter (partial < 0))
|
||||||
|
count)
|
||||||
|
|
||||||
;; part 2 solution
|
;; part 2 solution
|
||||||
(->> puzzles
|
(->> puzzles
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user