2019: day02: part 1

This commit is contained in:
Antoine Martin 2019-12-02 11:50:39 +01:00
parent 9903508129
commit c59cd75be3
4 changed files with 86 additions and 1 deletions

1
aoc2019/input/day02.txt Normal file
View file

@ -0,0 +1 @@
1,0,0,3,1,1,2,3,1,3,4,3,1,5,0,3,2,9,1,19,1,19,5,23,1,9,23,27,2,27,6,31,1,5,31,35,2,9,35,39,2,6,39,43,2,43,13,47,2,13,47,51,1,10,51,55,1,9,55,59,1,6,59,63,2,63,9,67,1,67,6,71,1,71,13,75,1,6,75,79,1,9,79,83,2,9,83,87,1,87,6,91,1,91,13,95,2,6,95,99,1,10,99,103,2,103,9,107,1,6,107,111,1,10,111,115,2,6,115,119,1,5,119,123,1,123,13,127,1,127,5,131,1,6,131,135,2,135,13,139,1,139,2,143,1,143,10,0,99,2,0,14,0

82
aoc2019/src/day02.rs Normal file
View file

@ -0,0 +1,82 @@
use aoc::err;
use aoc::Result;
const INPUT: &str = include_str!("../input/day02.txt");
fn parse_intcode(input: &str) -> Result<Vec<usize>> {
input
.trim_end()
.split(',')
.map(|x| x.parse().map_err(|e| err!("couldn't parse int: {}", e)))
.collect()
}
pub fn run() -> Result<()> {
let intcode = parse_intcode(INPUT)?;
println!("part 1: {}", part1(&mut intcode.clone())?);
Ok(())
}
fn eval(intcode: &mut [usize]) -> Result<()> {
let mut pc = 0;
while intcode[pc] != 99 {
let op1 = intcode[pc + 1];
let op2 = intcode[pc + 2];
let res = intcode[pc + 3];
match intcode[pc] {
1 => intcode[res] = intcode[op1] + intcode[op2],
2 => intcode[res] = intcode[op1] * intcode[op2],
_ => return Err(err!("unknown opcode: {}", intcode[pc])),
};
pc += 4;
}
Ok(())
}
fn part1(input: &mut Vec<usize>) -> Result<usize> {
input[1] = 12;
input[2] = 2;
eval(input)?;
Ok(input[0])
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn part1_provided() {
let mut intcode = parse_intcode("1,9,10,3,2,3,11,0,99,30,40,50").unwrap();
eval(&mut intcode).unwrap();
assert_eq!(intcode, &[3500, 9, 10, 70, 2, 3, 11, 0, 99, 30, 40, 50]);
let mut intcode = parse_intcode("1,0,0,0,99").unwrap();
eval(&mut intcode).unwrap();
assert_eq!(intcode, &[2, 0, 0, 0, 99]);
let mut intcode = parse_intcode("2,3,0,3,99").unwrap();
eval(&mut intcode).unwrap();
assert_eq!(intcode, &[2, 3, 0, 6, 99]);
let mut intcode = parse_intcode("2,4,4,5,99,0").unwrap();
eval(&mut intcode).unwrap();
assert_eq!(intcode, &[2, 4, 4, 5, 99, 9801]);
let mut intcode = parse_intcode("1,1,1,4,99,5,6,0,99").unwrap();
eval(&mut intcode).unwrap();
assert_eq!(intcode, &[30, 1, 1, 4, 2, 5, 6, 0, 99]);
}
#[test]
fn part1_real() {
let mut intcode = parse_intcode(INPUT).unwrap();
assert_eq!(part1(&mut intcode).unwrap(), 6568671);
}
}

View file

@ -1,4 +1,5 @@
pub mod day01;
pub mod day02;
pub type Result<T> = std::result::Result<T, Box<dyn std::error::Error>>;

View file

@ -1,11 +1,12 @@
use std::env;
use aoc2019::day01;
use aoc2019::day02;
use aoc2019::Result;
fn main() -> Result<()> {
let days: &[fn() -> Result<()>] = &[day01::run];
let days: &[fn() -> Result<()>] = &[day01::run, day02::run];
let mut args = env::args();
args.next();