advent-of-code/04/src/main.rs

104 lines
2.6 KiB
Rust
Raw Normal View History

2018-12-04 18:36:08 +01:00
#[macro_use] extern crate text_io;
use std::{fs, cmp};
enum GuardAction {
BeginShift(i32),
FallAsleep,
WakeUp,
2018-12-04 18:36:08 +01:00
}
struct Guard {
events: Vec<Event>,
id: i32,
}
#[derive(Eq)]
struct DateTime {
month: i32,
day: i32,
hour: i32,
minute: i32,
sortable_time: i32,
}
// Calculate the absolute minute count relative to 01/01 0:00. Months are assumed to have 31
// days because we just want to ensure that a higher month always results in a higher
// timestamp. Think of this as a primitive and less correct unix epoch.
pub fn get_sortable_time(month: i32, day: i32, hour: i32, minute: i32) -> i32 {
return minute + hour * 60 + day * 60 * 24 + month * 60 * 24 * 31;
}
impl DateTime {
pub fn new(month: i32, day: i32, hour: i32, minute: i32) -> Self {
DateTime {
month,
day,
hour,
minute,
sortable_time: get_sortable_time(month, day, hour, minute)
}
}
}
impl Ord for DateTime {
fn cmp(&self, other: &DateTime) -> cmp::Ordering {
self.sortable_time.cmp(&other.sortable_time)
}
}
impl PartialOrd for DateTime {
fn partial_cmp(&self, other: &DateTime) -> Option<cmp::Ordering> {
Some(other.cmp(self))
}
}
impl PartialEq for DateTime {
fn eq(&self, other: &DateTime) -> bool {
self.sortable_time == other.sortable_time
}
}
struct Event {
datetime: DateTime,
action: GuardAction,
}
impl Event {
pub fn new(datetime: DateTime, action: GuardAction) -> Self {
Event {
datetime,
action
}
}
}
fn parse_action(input: &String) -> GuardAction {
match input[0] {
"f" => GuardAction::FallAsleep,
"w" => GuardAction::WakeUp,
"G" => {
let gid: i32;
scan!(input.bytes() => "Guard #{} begins shift", gid);
GuardAction::BeginShift(gid)
}
_ => // undefined?
}
2018-12-04 18:36:08 +01:00
}
fn event_from_line(line: &String) -> Event {
let (month, day, hour, minute, unparsed_action): (i32, i32, i32, i32, String);
scan!(line.bytes() => "[1518-{}-{} {}:{}] {}", month, day, hour, minute, unparsed_action);
let datetime = DateTime::new(month, day, hour, minute);
return Event::new(datetime, parse_action(&unparsed_action));
}
fn main() {
let file = fs::read_to_string("../input").expect("can’t access file");
let lines: Vec<&str> = file.split("\n").collect();
let mut events: Vec<Event> = Vec::new();
for line in lines {
events.push(event_from_line(&line.to_string()));
}
}