rust(day16): Implement day 16
This commit is contained in:
parent
87fdcce197
commit
1271eb0e98
148
rust/src/main.rs
148
rust/src/main.rs
|
@ -518,6 +518,153 @@ mod day15 {
|
|||
}
|
||||
}
|
||||
|
||||
mod day16 {
|
||||
use std::collections::BTreeSet;
|
||||
use std::fmt;
|
||||
use std::time::{Duration, Instant};
|
||||
|
||||
type Ticket = Vec<usize>;
|
||||
|
||||
#[derive(PartialEq, Eq, Hash)]
|
||||
struct Range {
|
||||
from: usize,
|
||||
to: usize,
|
||||
}
|
||||
|
||||
impl fmt::Debug for Range {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
f.write_fmt(format_args!("{} - {}", self.from, self.to))
|
||||
}
|
||||
}
|
||||
impl Range {
|
||||
fn from(s: &str) -> Self {
|
||||
let mut range = s.split("-");
|
||||
Self {
|
||||
from: range.next().unwrap().parse().unwrap(),
|
||||
to: range.next().unwrap().parse().unwrap(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, PartialEq, Eq, Hash)]
|
||||
struct Rule {
|
||||
name: String,
|
||||
ranges: (Range, Range),
|
||||
}
|
||||
|
||||
pub fn run(print: bool) -> Duration {
|
||||
let input = std::fs::read_to_string("../inputs/16").unwrap();
|
||||
let instant = Instant::now();
|
||||
|
||||
let mut lines = input.lines();
|
||||
let mut rules = Vec::new();
|
||||
while let Some(line) = lines.next() {
|
||||
let mut colon_split = line.split(": ");
|
||||
let name = colon_split.next().unwrap().to_string();
|
||||
if name.is_empty() {
|
||||
break;
|
||||
}
|
||||
|
||||
let mut ranges = colon_split.next().unwrap().split(" or ");
|
||||
|
||||
rules.push(Rule {
|
||||
name,
|
||||
ranges: (
|
||||
Range::from(ranges.next().unwrap()),
|
||||
Range::from(ranges.next().unwrap()),
|
||||
),
|
||||
});
|
||||
}
|
||||
|
||||
lines.next(); // "your ticket:"
|
||||
|
||||
let your_ticket: Ticket = lines
|
||||
.next()
|
||||
.unwrap()
|
||||
.split(",")
|
||||
.flat_map(str::parse)
|
||||
.collect();
|
||||
|
||||
lines.next(); // ""
|
||||
lines.next(); // "nearby tickets:"
|
||||
|
||||
let mut nearby_tickets: Vec<Ticket> = Vec::new();
|
||||
while let Some(line) = lines.next() {
|
||||
nearby_tickets.push(line.split(",").flat_map(str::parse).collect());
|
||||
}
|
||||
|
||||
let mut part1 = 0;
|
||||
let mut valid_tickets = Vec::new();
|
||||
for ticket in &nearby_tickets {
|
||||
let mut valid = true;
|
||||
for field in ticket {
|
||||
let rule = rules.iter().find(|rule| {
|
||||
(field >= &rule.ranges.0.from && field <= &rule.ranges.0.to)
|
||||
|| (field >= &rule.ranges.1.from && field <= &rule.ranges.1.to)
|
||||
});
|
||||
if rule.is_none() {
|
||||
valid = false;
|
||||
part1 += field;
|
||||
}
|
||||
}
|
||||
if valid {
|
||||
valid_tickets.push(ticket);
|
||||
}
|
||||
}
|
||||
|
||||
if print {
|
||||
dbg!(part1);
|
||||
}
|
||||
|
||||
let mut field_possibilities: Vec<_> = (0..your_ticket.len())
|
||||
.map(|_| (0..rules.len()).collect::<BTreeSet<_>>())
|
||||
.collect();
|
||||
|
||||
for ticket in valid_tickets {
|
||||
for (field, possibilities) in ticket.iter().zip(field_possibilities.iter_mut()) {
|
||||
for rule_i in possibilities.clone().iter() {
|
||||
let rule = &rules[*rule_i];
|
||||
if !((field >= &rule.ranges.0.from && field <= &rule.ranges.0.to)
|
||||
|| (field >= &rule.ranges.1.from && field <= &rule.ranges.1.to))
|
||||
{
|
||||
possibilities.remove(&rule_i);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let mut part2 = 1;
|
||||
loop {
|
||||
let field_possibility = field_possibilities
|
||||
.iter()
|
||||
.enumerate()
|
||||
.find(|(_, fp)| fp.len() == 1);
|
||||
|
||||
let (i, possibility) = match field_possibility {
|
||||
Some((i, v)) => (i, *v.iter().next().take().unwrap()),
|
||||
None => break,
|
||||
};
|
||||
|
||||
for fp in field_possibilities.iter_mut() {
|
||||
fp.remove(&possibility);
|
||||
}
|
||||
|
||||
let rule = &rules[possibility].name;
|
||||
//println!("{} is {}", rule, your_ticket[i]);
|
||||
|
||||
if rule.starts_with("departure") {
|
||||
part2 *= your_ticket[i];
|
||||
}
|
||||
}
|
||||
|
||||
if print {
|
||||
dbg!(part2);
|
||||
}
|
||||
|
||||
instant.elapsed()
|
||||
}
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let days = [
|
||||
day01::run,
|
||||
|
@ -535,6 +682,7 @@ fn main() {
|
|||
day13::run,
|
||||
day14::run,
|
||||
day15::run,
|
||||
day16::run,
|
||||
];
|
||||
|
||||
if let Some(day) = std::env::args().nth(1) {
|
||||
|
|
Loading…
Reference in New Issue