solution refactoring: day 1, with BinaryHeap<Reverse<T>>

This commit is contained in:
YK 2024-12-02 17:09:15 +03:00
parent 507f88f5d4
commit 005e655daf

View File

@ -1,47 +1,39 @@
use crate::prelude::*; use crate::prelude::*;
pub type I = (Vec<u32>, Vec<u32>); pub type I = (B<Rev<u32>>, B<Rev<u32>>);
pub type O = u32; pub type O = u32;
fn _parse (data: &str) -> I { fn _parse (data: &str) -> I {
let (mut l, mut r) = (vec![], vec![]); let (mut l, mut r) = (B::new(), B::new());
data.lines().for_each(|line| { data.lines().for_each(|line| {
let mut iter = line.split_whitespace(); let mut iter = line.split_whitespace();
let left = p!(n!(iter)); let left = p!(n!(iter));
let right = p!(n!(iter)); let right = p!(n!(iter));
l.push(left); l.push(Rev(left));
r.push(right); r.push(Rev(right));
}); });
(l, r) (l, r)
} }
fn _silver ((mut left, mut right): I) -> O { fn _silver ((left, right): I) -> O {
left.sort(); left.into_iter_sorted().zip(right.into_iter_sorted()).fold(0, |acc, (l, r)| ({ println!("{:?}", (acc, l, r)); 0 }) + acc + l.0.abs_diff(r.0))
right.sort();
left.into_iter().zip(right.into_iter()).fold(0, |acc, (l, r)| acc + l.abs_diff(r))
} }
fn _gold ((left, right): I) -> O { fn _gold ((left, right): I) -> O {
let right = right.into_iter().counts(); let right = right.into_iter().map(|e| e.0).counts();
left.into_iter().fold(0, |acc, cur| acc + cur * *right.get(&cur).unwrap_or(&0) as u32) left.into_iter().fold(0, |acc, cur| acc + cur.0 * *right.get(&cur.0).unwrap_or(&0) as u32)
} }
#[cfg(test)] #[cfg(test)]
mod test { mod test {
use super::*; use super::*;
fn read () -> I {
let data = inc!(1);
_parse(data)
}
#[test] #[test]
fn silver () { fn silver () {
let data = read(); let data = _parse(inc!(1));
let ans = _silver(data); let ans = _silver(data);
assert_eq!(ans, 2196996) assert_eq!(ans, 2196996)
@ -49,7 +41,7 @@ mod test {
#[test] #[test]
fn gold () { fn gold () {
let data = read(); let data = _parse(inc!(1));
let ans = _gold(data); let ans = _gold(data);
assert_eq!(ans, 23655822) assert_eq!(ans, 23655822)