1 Commits

Author SHA1 Message Date
30d1a16075 Collect into vec for nicer vectorization 2022-12-02 09:52:09 +01:00
2 changed files with 25 additions and 34 deletions

View File

@@ -2,7 +2,6 @@ use std::ops::Add;
use anyhow::Result; use anyhow::Result;
use nom::character::complete::newline; use nom::character::complete::newline;
use nom::combinator::opt;
use nom::multi::separated_list0; use nom::multi::separated_list0;
use nom::sequence::terminated; use nom::sequence::terminated;
use nom::IResult; use nom::IResult;
@@ -14,13 +13,9 @@ fn parse_elf(input: &[u8]) -> IResult<&[u8], i32> {
reduce_many1(terminated(nom::character::complete::i32, newline), Add::add)(input) reduce_many1(terminated(nom::character::complete::i32, newline), Add::add)(input)
} }
fn parse_max(input: &[u8]) -> IResult<&[u8], i32> {
reduce_many1(terminated(parse_elf, opt(newline)), Ord::max)(input)
}
pub fn part1(input: &[u8]) -> Result<String> { pub fn part1(input: &[u8]) -> Result<String> {
let result = parse_input(input, parse_max)?.to_string(); let elves = parse_input(input, parse_elf_list)?;
Ok(result) Ok(elves.into_iter().fold(0, Ord::max).to_string())
} }
fn parse_elf_list(input: &[u8]) -> IResult<&[u8], Vec<i32>> { fn parse_elf_list(input: &[u8]) -> IResult<&[u8], Vec<i32>> {

View File

@@ -1,15 +1,12 @@
use std::ops::Add;
use anyhow::Result; use anyhow::Result;
use nom::character::complete::newline; use nom::character::complete::newline;
use nom::combinator::map;
use nom::combinator::map_res; use nom::combinator::map_res;
use nom::multi::many0;
use nom::sequence::separated_pair; use nom::sequence::separated_pair;
use nom::sequence::terminated; use nom::sequence::terminated;
use nom::IResult; use nom::IResult;
use crate::common::parse_input; use crate::common::parse_input;
use crate::common::reduce_many1;
#[derive(Copy, Clone, Eq, PartialEq)] #[derive(Copy, Clone, Eq, PartialEq)]
enum Rps { enum Rps {
@@ -73,7 +70,7 @@ impl TryFrom<u8> for Rps {
} }
} }
fn parse_line(input: &[u8]) -> IResult<&[u8], (Rps, Rps)> { fn parse_plan(input: &[u8]) -> IResult<&[u8], Vec<(Rps, Rps)>> {
fn parse_rps(input: &[u8]) -> IResult<&[u8], Rps> { fn parse_rps(input: &[u8]) -> IResult<&[u8], Rps> {
// Note: alpha1 also sort of works but is significantly slower // Note: alpha1 also sort of works but is significantly slower
map_res(nom::bytes::complete::take(1usize), |v: &[u8]| { map_res(nom::bytes::complete::take(1usize), |v: &[u8]| {
@@ -81,34 +78,33 @@ fn parse_line(input: &[u8]) -> IResult<&[u8], (Rps, Rps)> {
})(input) })(input)
} }
terminated( fn parse_line(input: &[u8]) -> IResult<&[u8], (Rps, Rps)> {
separated_pair(parse_rps, nom::character::complete::char(' '), parse_rps), separated_pair(parse_rps, nom::character::complete::char(' '), parse_rps)(input)
newline, }
)(input)
many0(terminated(parse_line, newline))(input)
} }
pub fn part1(input: &[u8]) -> Result<String> { pub fn part1(input: &[u8]) -> Result<String> {
parse_input( let plan = parse_input(input, parse_plan)?;
input,
reduce_many1( let result: u32 = plan
map(parse_line, |(them, us)| us.score() + us.score_against(them)), .into_iter()
Add::add, .map(|(them, us)| us.score() + us.score_against(them))
), .sum();
)
.map(|sum| sum.to_string()) Ok(result.to_string())
} }
pub fn part2(input: &[u8]) -> Result<String> { pub fn part2(input: &[u8]) -> Result<String> {
parse_input( let plan = parse_input(input, parse_plan)?;
input,
reduce_many1( let result: u32 = plan
map(parse_line, |(them, us)| { .into_iter()
us.score_result() + us.needed(them).score() .map(|(them, us)| us.score_result() + us.needed(them).score())
}), .sum();
Add::add,
), Ok(result.to_string())
)
.map(|sum| sum.to_string())
} }
#[cfg(test)] #[cfg(test)]