From b1896185c1c8a17a16b6b6d83dbc41d52a5de2b4 Mon Sep 17 00:00:00 2001
From: mirsella <mirsella@protonmail.com>
Date: Thu, 5 Dec 2024 08:34:52 +0100
Subject: [PATCH] make code cleaner

---
 2024/day5/src/main.rs | 22 +++++++++-------------
 1 file changed, 9 insertions(+), 13 deletions(-)

diff --git a/2024/day5/src/main.rs b/2024/day5/src/main.rs
index 410dcae..a069a97 100644
--- a/2024/day5/src/main.rs
+++ b/2024/day5/src/main.rs
@@ -7,9 +7,11 @@ fn part1(input: &str) -> usize {
         let mut map = HashMap::<usize, Vec<usize>>::new();
         split.0.lines().for_each(|l| {
             let s = l.split_once('|').unwrap();
+            let value = s.0.parse::<usize>().unwrap();
             map.entry(s.1.parse::<usize>().unwrap())
                 .or_default()
-                .push(s.0.parse::<usize>().unwrap());
+                .push(value);
+            map.entry(value).or_default();
         });
         map
     };
@@ -21,7 +23,7 @@ fn part1(input: &str) -> usize {
     updates
         .filter_map(|update| {
             update
-                .is_sorted_by(|a, b| ordering.get(b).map(|v| v.contains(a)).unwrap_or_default())
+                .is_sorted_by(|a, b| ordering[b].contains(a))
                 .then_some(update[update.len() / 2])
         })
         .sum()
@@ -32,9 +34,11 @@ fn part2(input: &str) -> usize {
         let mut map = HashMap::<usize, Vec<usize>>::new();
         split.0.lines().for_each(|l| {
             let s = l.split_once('|').unwrap();
+            let value = s.0.parse::<usize>().unwrap();
             map.entry(s.1.parse::<usize>().unwrap())
                 .or_default()
-                .push(s.0.parse::<usize>().unwrap());
+                .push(value);
+            map.entry(value).or_default();
         });
         map
     };
@@ -45,16 +49,8 @@ fn part2(input: &str) -> usize {
     });
     updates
         .filter_map(|mut update| {
-            if !update
-                .is_sorted_by(|a, b| ordering.get(b).map(|v| v.contains(a)).unwrap_or_default())
-            {
-                update.sort_by(|a, b| {
-                    ordering
-                        .get(b)
-                        .map(|v| v.contains(a))
-                        .unwrap_or_default()
-                        .cmp(&true)
-                });
+            if !update.is_sorted_by(|a, b| ordering[b].contains(a)) {
+                update.sort_by(|a, b| ordering[b].contains(a).cmp(&true));
                 return Some(update[update.len() / 2]);
             }
             None