summaryrefslogtreecommitdiff
path: root/2023/rust/day4/src/lib.rs
diff options
context:
space:
mode:
Diffstat (limited to '2023/rust/day4/src/lib.rs')
-rw-r--r--2023/rust/day4/src/lib.rs132
1 files changed, 132 insertions, 0 deletions
diff --git a/2023/rust/day4/src/lib.rs b/2023/rust/day4/src/lib.rs
new file mode 100644
index 0000000..ca38b56
--- /dev/null
+++ b/2023/rust/day4/src/lib.rs
@@ -0,0 +1,132 @@
+fn parse(input: &str) -> Vec<String> {
+ input
+ .trim_end()
+ .split('\n')
+ .map(|x| {
+ let b = x.to_string();
+ let k = b.split(':').last().unwrap();
+
+ return k.trim().to_string();
+ })
+ .collect()
+}
+
+pub fn part1(input: &str) -> u32 {
+ let mut res: u32 = 0;
+
+ let cards: Vec<String> = parse(input);
+ for card in cards {
+ let numbers: Vec<String> = card.split('|').map(|x| x.trim().to_string()).collect();
+ assert_eq!(numbers.len(), 2);
+
+ let nums: Vec<Vec<_>> = numbers
+ .iter()
+ .map(|x| {
+ x.replace(" ", " ")
+ .split(' ')
+ .map(|x| x.trim().parse::<u32>().unwrap())
+ .collect()
+ })
+ .collect();
+
+ assert_eq!(nums.len(), 2);
+ let (winnings, plays) = (&nums[0], &nums[1]);
+
+ let mut k = 0.5;
+ for win in winnings {
+ if plays.contains(&win) {
+ k *= 2.;
+ }
+ }
+ res += k as u32;
+ }
+
+ res
+}
+
+//
+//
+//
+// 1 -> 4 | x
+// 2 -> 2 | y x
+// 3 -> 2 | y y y x
+// 4 -> 1 | y y y y y y y x
+// 5 -> 0 | y y y y y y y y y y y y y x
+// 6 -> 0 | x
+//
+//
+pub fn part2(input: &str) -> u32 {
+ let mut res: u32 = 0;
+
+ let cards: Vec<String> = parse(input);
+ let mut incr: Vec<u32> = Vec::new();
+ incr.resize_with(cards.len(), Default::default);
+
+ let mut i = 0;
+ for card in cards {
+ let numbers: Vec<String> = card.split('|').map(|x| x.trim().to_string()).collect();
+ assert_eq!(numbers.len(), 2);
+
+ let nums: Vec<Vec<_>> = numbers
+ .iter()
+ .map(|x| {
+ x.replace(" ", " ")
+ .split(' ')
+ .map(|x| x.trim().parse::<u32>().unwrap())
+ .collect()
+ })
+ .collect();
+
+ assert_eq!(nums.len(), 2);
+ let (winnings, plays) = (&nums[0], &nums[1]);
+
+ let mut k = 0;
+ for win in winnings {
+ if plays.contains(&win) {
+ k += 1;
+ }
+ }
+
+ for j in (i + 1)..(i + 1 + k) {
+ incr[j] += 1 + incr[i];
+ }
+
+ res += 1 + incr[i];
+ i += 1;
+ }
+
+ res
+}
+
+#[cfg(test)]
+mod tests {
+ use super::*;
+
+ #[test]
+ fn example_part1() {
+ let input = include_str!("../example.txt");
+ let result = part1(input);
+ assert_eq!(result, 13);
+ }
+
+ #[test]
+ fn input_part1() {
+ let input = include_str!("../input.txt");
+ let result = part1(input);
+ assert_eq!(result, 25174);
+ }
+
+ #[test]
+ fn example_part2() {
+ let input = include_str!("../example.txt");
+ let result = part2(input);
+ assert_eq!(result, 30);
+ }
+
+ #[test]
+ fn input_part2() {
+ let input = include_str!("../input.txt");
+ let result = part2(input);
+ assert_eq!(result, 6420979);
+ }
+}