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 39c5eb1897

View File

@ -1,47 +1,39 @@
use crate::prelude::*;
pub type I = (Vec<u32>, Vec<u32>);
pub type I = (B<Rev<u32>>, B<Rev<u32>>);
pub type O = u32;
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| {
let mut iter = line.split_whitespace();
let left = p!(n!(iter));
let right = p!(n!(iter));
l.push(left);
r.push(right);
l.push(Rev(left));
r.push(Rev(right));
});
(l, r)
}
fn _silver ((mut left, mut right): I) -> O {
left.sort();
right.sort();
left.into_iter().zip(right.into_iter()).fold(0, |acc, (l, r)| acc + l.abs_diff(r))
fn _silver ((left, right): I) -> O {
left.into_iter_sorted().zip(right.into_iter_sorted()).fold(0, |acc, (l, r)| acc + l.0.abs_diff(r.0))
}
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)]
mod test {
use super::*;
fn read () -> I {
let data = inc!(1);
_parse(data)
}
#[test]
fn silver () {
let data = read();
let data = _parse(inc!(1));
let ans = _silver(data);
assert_eq!(ans, 2196996)
@ -49,7 +41,7 @@ mod test {
#[test]
fn gold () {
let data = read();
let data = _parse(inc!(1));
let ans = _gold(data);
assert_eq!(ans, 23655822)