74 lines
1.8 KiB
OCaml
74 lines
1.8 KiB
OCaml
#use "lib/file_utils.ml" ;;
|
|
|
|
type shape = Rock | Paper | Scissors
|
|
type outcome = Win | Tie | Loss
|
|
|
|
let shape_of_char = function
|
|
| 'A' | 'X' -> Rock
|
|
| 'B' | 'Y' -> Paper
|
|
| 'C' | 'Z' -> Scissors
|
|
| s -> failwith ("invalid shape: " ^ (String.make 1 s))
|
|
|
|
let score_of_shape = function
|
|
| Rock -> 1
|
|
| Paper -> 2
|
|
| Scissors -> 3
|
|
|
|
let score_of_outcome = function
|
|
| Loss -> 0
|
|
| Tie -> 3
|
|
| Win -> 6
|
|
|
|
let part1 line =
|
|
let outcome_of_round = function
|
|
| Rock, Scissors -> Win
|
|
| Paper, Rock -> Win
|
|
| Scissors, Paper -> Win
|
|
| a, b when a = b -> Tie
|
|
| _ -> Loss
|
|
in
|
|
let op_move = shape_of_char (String.get line 0) in
|
|
let my_move = shape_of_char (String.get line 2) in
|
|
let outcome = outcome_of_round(my_move, op_move) in
|
|
let outcome_score = score_of_outcome outcome in
|
|
let shape_score = score_of_shape my_move in
|
|
outcome_score + shape_score
|
|
|
|
let part2 line =
|
|
let outcome_of_char = function
|
|
| 'X' -> Loss
|
|
| 'Y' -> Tie
|
|
| 'Z' -> Win
|
|
| o -> failwith ("invalid outcome: " ^ (String.make 1 o))
|
|
in
|
|
let make_move = function
|
|
| Rock, Win -> Paper
|
|
| Rock, Loss -> Scissors
|
|
| Paper, Win -> Scissors
|
|
| Paper, Loss -> Rock
|
|
| Scissors, Win -> Rock
|
|
| Scissors, Loss -> Paper
|
|
| s, Tie -> s
|
|
in
|
|
let op_move = shape_of_char (String.get line 0) in
|
|
let outcome = outcome_of_char (String.get line 2) in
|
|
let my_move = make_move(op_move, outcome) in
|
|
let outcome_score = score_of_outcome outcome in
|
|
let shape_score = score_of_shape my_move in
|
|
outcome_score + shape_score
|
|
|
|
let play lines part rules =
|
|
let score = lines
|
|
|> List.map rules
|
|
|> List.fold_left (+) 0
|
|
in
|
|
Printf.printf "Part %d: %d\n" part score
|
|
;;
|
|
|
|
let contents = read_file "inputs/2.txt" in
|
|
let lines = String.split_on_char '\n' contents
|
|
|> List.filter (fun l -> String.trim l <> "")
|
|
in
|
|
play lines 1 part1;
|
|
play lines 2 part2
|