[AoC2022] Rust 02-02

1 files changed, 67 insertions, 15 deletions

diff --git a/AoC2022/02/solver.rs b/AoC2022/02/solver.rs
index 1f9a394..2fd1e62 100644
--- a/AoC2022/02/solver.rs
+++ b/AoC2022/02/solver.rs
@@ -1,7 +1,7 @@
 use std::fs;
 use std::io::{self, BufRead};
 
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 enum Hand {
     Rock,
     Paper,
@@ -25,28 +25,80 @@ fn weight(play: Hand) -> u32 {
     }
 }
 
+enum Conclusion {
+    Loose,
+    Draw,
+    Win,
+}
+
+fn strategy(play: &str) -> Conclusion {
+    match play {
+        "X" => Conclusion::Loose,
+        "Y" => Conclusion::Draw,
+        "Z" => Conclusion::Win,
+        &_ => panic!("Invalid input"),
+    }
+}
+fn wins(play: Hand) -> Hand {
+    match play {
+        Hand::Rock => Hand::Scissors,
+        Hand::Scissors => Hand::Paper,
+        Hand::Paper => Hand::Rock,
+    }
+}
+fn looses(play: Hand) -> Hand {
+    match play {
+        Hand::Scissors => Hand::Rock,
+        Hand::Paper => Hand::Scissors,
+        Hand::Rock => Hand::Paper,
+    }
+}
+
+fn pick_move(strategy: Conclusion, other: Hand) -> Hand {
+    match strategy {
+        Conclusion::Draw => other,
+        Conclusion::Win => looses(other),
+        Conclusion::Loose => wins(other),
+    }
+}
+
 fn fight(my: Hand, other: Hand) -> u32 {
     match (my, other) {
         (a, b) if a == b => 3,
-        (Hand::Rock, Hand::Scissors)
-        | (Hand::Scissors, Hand::Paper)
-        | (Hand::Paper, Hand::Rock) => 6,
-        _ => 0,
+        (a, b) => {
+            if wins(a) == b {
+                6
+            } else {
+                0
+            }
+        }
     }
 }
 
-fn main() -> std::io::Result<()> {
+fn fixed_plays(my: &str, _oponent: Hand) -> Hand {
+    translate(my)
+}
+
+fn reactive_plays(my: &str, oponent: Hand) -> Hand {
+    pick_move(strategy(my), oponent)
+}
+
+fn solver(strategy: &dyn Fn(&str, Hand) -> Hand) -> u32 {
     let file = fs::File::open("input").unwrap();
     let lines = io::BufReader::new(file).lines();
-    let mut score = 0;
-    for line in lines {
-        if let [ot, my] = line?.split_whitespace().collect::<Vec<_>>()[..] {
-            score += fight(translate(my), translate(ot)) + weight(translate(my));
+    lines.fold(0, |acc, line| {
+        if let [ot, my] = line.unwrap().split_whitespace().collect::<Vec<_>>()[..] {
+            let opo = translate(ot);
+            let my_hand = strategy(my, opo.clone());
+            acc + fight(my_hand.clone(), opo) + weight(my_hand)
         } else {
-            todo!()
-        };
-    }
+            panic!("Wrong line")
+        }
+    })
+}
 
-    println!("{}", score);
-    Ok(())
+fn main() {
+    assert_eq!(12535, solver(&fixed_plays));
+    assert_eq!(15457, solver(&reactive_plays));
+    println!("All test passed.")
 }