67 lines
1.5 KiB
Rust
67 lines
1.5 KiB
Rust
#![warn(clippy::pedantic)]
|
|
|
|
use std::{
|
|
io::{stdin, Read},
|
|
iter,
|
|
str::FromStr,
|
|
};
|
|
|
|
enum Instruction {
|
|
Noop,
|
|
Addx(i32),
|
|
}
|
|
|
|
impl FromStr for Instruction {
|
|
type Err = ();
|
|
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
let mut args = s.split_whitespace();
|
|
match args.next().ok_or(())? {
|
|
"noop" => Ok(Self::Noop),
|
|
"addx" => {
|
|
let i = args.next().ok_or(())?.parse().map_err(|_| ())?;
|
|
Ok(Self::Addx(i))
|
|
}
|
|
_ => Err(()),
|
|
}
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
let mut data = String::new();
|
|
stdin().read_to_string(&mut data).unwrap();
|
|
|
|
let values = data
|
|
.lines()
|
|
.map(|l| l.parse().unwrap())
|
|
.flat_map(|i: Instruction| match i {
|
|
Instruction::Noop => vec![0],
|
|
Instruction::Addx(i) => vec![0, i],
|
|
})
|
|
.scan(1, |acc, i| {
|
|
*acc += i;
|
|
Some(*acc)
|
|
});
|
|
let values = iter::once(1)
|
|
.chain(values)
|
|
.enumerate()
|
|
.map(|(i, x)| (i32::try_from(i).unwrap() + 1, x));
|
|
|
|
let sum: i32 = values
|
|
.clone()
|
|
.skip(20 - 1)
|
|
.step_by(40)
|
|
.map(|(i, x)| i * x)
|
|
.sum();
|
|
println!("{:?}", sum);
|
|
|
|
let crt: Vec<_> = values
|
|
.map(|(i, x)| ((i - 1) % 40).abs_diff(x) < 2)
|
|
.map(|x| if x { '#' } else { '.' })
|
|
.collect();
|
|
|
|
for line in crt.chunks_exact(40) {
|
|
println!("{}", line.iter().collect::<String>());
|
|
}
|
|
}
|