Compare commits

...

9 Commits

Author SHA1 Message Date
272761b05c day16 2024-01-02 10:48:52 -05:00
f9d58356cd day15 2023-12-29 11:30:39 -05:00
a0c9ce9882 day14 2023-12-29 10:32:23 -05:00
9fc94b0e06 day13 2023-12-28 16:42:04 -05:00
c88b97182d day12 2023-12-28 15:43:02 -05:00
0407edf928 day10 2023-12-28 13:36:54 -05:00
b44361e14d day9 2023-12-28 13:35:58 -05:00
90561f8fa9 day8 2023-12-28 13:35:39 -05:00
2c499e9efe day7 2023-12-28 13:35:25 -05:00
12 changed files with 1113 additions and 0 deletions

52
Cargo.lock generated
View File

@@ -9,10 +9,21 @@ dependencies = [
"aoc-runner",
"aoc-runner-derive",
"itertools",
"lazy_static",
"regex",
"strum",
"strum_macros",
]
[[package]]
name = "aho-corasick"
version = "1.1.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b2969dcb958b36655471fc61f7e416fa76033bdd4bfed0678d8fee1e2d07a1f0"
dependencies = [
"memchr",
]
[[package]]
name = "aoc-runner"
version = "0.3.0"
@@ -69,6 +80,18 @@ version = "1.0.10"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b1a46d1a171d865aa5f83f92695765caa047a9b4cbae2cbf37dbd613a793fd4c"
[[package]]
name = "lazy_static"
version = "1.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646"
[[package]]
name = "memchr"
version = "2.6.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f665ee40bc4a3c5590afb1e9677db74a508659dfd71e126420da8274909a0167"
[[package]]
name = "proc-macro2"
version = "1.0.71"
@@ -87,6 +110,35 @@ dependencies = [
"proc-macro2",
]
[[package]]
name = "regex"
version = "1.10.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "380b951a9c5e80ddfd6136919eef32310721aa4aacd4889a8d39124b026ab343"
dependencies = [
"aho-corasick",
"memchr",
"regex-automata",
"regex-syntax",
]
[[package]]
name = "regex-automata"
version = "0.4.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5f804c7828047e88b2d32e2d7fe5a105da8ee3264f01902f796c8e067dc2483f"
dependencies = [
"aho-corasick",
"memchr",
"regex-syntax",
]
[[package]]
name = "regex-syntax"
version = "0.8.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c08c74e62047bb2de4ff487b251e4a92e24f48745648451635cec7d591162d9f"
[[package]]
name = "rustversion"
version = "1.0.14"

View File

@@ -14,3 +14,5 @@ aoc-runner-derive = "0.3.0"
itertools = "0.12.0"
strum = "0.25.0"
strum_macros = "0.25"
lazy_static = "1.4.0"
regex = "1.10.2"

109
src/day10.rs Normal file
View File

@@ -0,0 +1,109 @@
use std::collections::HashMap;
use aoc_runner_derive::{aoc, aoc_generator};
use itertools::Itertools;
#[aoc_generator(day10)]
fn parse(input: &str) -> Vec<u32> {
input.lines().map(str::parse).map(Result::unwrap).sorted().collect_vec()
}
#[aoc(day10, part1)]
fn part1(input: &Vec<u32>) -> u32 {
let mut count1 = 1;
let mut count3 = 1;
for (lhs, rhs) in input.iter().tuple_windows() {
match rhs - lhs {
1 => count1 += 1,
3 => count3 += 1,
_ => unreachable!()
}
}
count1 * count3
}
#[aoc(day10, part2)]
fn part2(input: &Vec<u32>) -> i64 {
let mut ans = HashMap::new();
ans.insert(0, 1);
for adapter in input.iter().cloned().map(|x| x as i32) {
let to_insert = ans.get(&(adapter as i32 - 1)).unwrap_or(&0)
+ ans.get(&(adapter as i32 - 2)).unwrap_or(&0)
+ ans.get(&(adapter as i32 - 3)).unwrap_or(&0);
ans.insert(adapter, to_insert.clone());
}
ans[&(*input.last().unwrap() as i32)]
}
#[cfg(test)]
mod tests {
use super::*;
const EX: &str = r"16
10
15
5
1
11
7
19
6
12
4";
const EX_2: &str = r"28
33
18
42
31
14
46
20
48
47
24
23
49
45
19
38
39
11
1
32
25
35
8
17
7
9
4
2
34
10
3";
#[test]
fn part1_example() {
assert_eq!(part1(&parse(EX)), 35);
}
#[test]
fn part1_example2() {
assert_eq!(part1(&parse(EX_2)), 220);
}
#[test]
fn part2_example() {
assert_eq!(part2(&parse(EX)), 8);
}
#[test]
fn part2_example2() {
assert_eq!(part2(&parse(EX_2)), 19208);
}
}

137
src/day12.rs Normal file
View File

@@ -0,0 +1,137 @@
use aoc_runner_derive::{aoc, aoc_generator};
use Instruction::*;
use itertools::Itertools;
#[derive(Debug)]
enum Instruction {
MoveNorth(i32),
MoveSouth(i32),
MoveEast(i32),
MoveWest(i32),
TurnLeft(i32),
TurnRight(i32),
MoveForward(i32)
}
impl From<(char, i32)> for Instruction {
fn from(value: (char, i32)) -> Self {
match value {
(i, v) if i == 'N' => MoveNorth(v),
(i, v) if i == 'S' => MoveSouth(v),
(i, v) if i == 'E' => MoveEast(v),
(i, v) if i == 'W' => MoveWest(v),
(i, v) if i == 'L' => TurnLeft(v),
(i, v) if i == 'R' => TurnRight(v),
(i, v) if i == 'F' => MoveForward(v),
_ => unreachable!()
}
}
}
#[aoc_generator(day12)]
fn parse(input: &str) -> Vec<Instruction> {
input
.lines()
.map(|line| {
(line.chars().next().unwrap(), line[1..].parse::<i32>().unwrap()).into()
})
.collect_vec()
}
const DIR_POS_MODIFIER: [(i32, i32); 4] = [(1, 0), (0, -1), (-1, 0), (0, 1)];
#[aoc(day12, part1)]
fn part1(input: &Vec<Instruction>) -> i32 {
let mut x = 0;
let mut y = 0;
let mut direction = 0; // 0 is east, thus 90 is south, 180 is west, and 270 is north
for instr in input {
match instr {
MoveNorth(val) => y += val,
MoveSouth(val) => y -= val,
MoveEast(val) => x += val,
MoveWest(val) => x -= val,
TurnLeft(val) => {
direction -= val;
if direction < 0 {
direction += 360;
}
},
TurnRight(val) => {
direction += val;
if direction >= 360 {
direction -= 360
}
},
MoveForward(val) => {
let (xmod, ymod) = DIR_POS_MODIFIER[(direction / 90) as usize];
x += xmod * val;
y += ymod * val;
},
}
}
x.abs() + y.abs()
}
#[aoc(day12, part2)]
fn part2(input: &Vec<Instruction>) -> i32 {
let mut way_x = 10;
let mut way_y = 1;
let mut ship_x: i32 = 0;
let mut ship_y: i32 = 0;
for instr in input {
match instr {
MoveNorth(val) => way_y += val,
MoveSouth(val) => way_y -= val,
MoveEast(val) => way_x += val,
MoveWest(val) => way_x -= val,
TurnLeft(val) => {
for _ in 0..(val / 90) {
let tmp = way_y;
way_y = way_x;
way_x = -tmp;
}
},
TurnRight(val) => {
for _ in 0..(val / 90) {
let tmp = way_y;
way_y = -way_x;
way_x = tmp;
}
},
MoveForward(val) => {
ship_x += way_x * val;
ship_y += way_y * val;
},
}
}
ship_x.abs() + ship_y.abs()
}
#[cfg(test)]
mod tests {
use super::*;
const EX: &str = r"F10
N3
F7
R90
F11";
#[test]
fn part1_example() {
assert_eq!(part1(&parse(EX)), 25);
}
#[test]
fn part2_example() {
assert_eq!(part2(&parse(EX)), 286);
}
}

76
src/day13.rs Normal file
View File

@@ -0,0 +1,76 @@
use aoc_runner_derive::{aoc, aoc_generator};
use itertools::Itertools;
#[aoc_generator(day13, part1)]
fn parse(input: &str) -> (i32, Vec<i32>) {
let (start, buses) = input
.lines()
.collect_tuple()
.unwrap();
let buses = buses
.split(',')
.filter_map(|num| num.parse::<i32>().ok())
.collect_vec();
(start.parse::<i32>().unwrap(), buses)
}
#[aoc(day13, part1)]
fn part1((start, buses): &(i32, Vec<i32>)) -> i32 {
let mut cur = 0;
loop {
if let Some(bus) = buses.iter().find(|x| (cur + start) % *x == 0) {
return bus * cur;
}
cur += 1;
}
}
#[aoc_generator(day13, part2)]
fn parse2(input: &str) -> Vec<Option<i64>> {
let buses = input
.lines().skip(1).next().unwrap();
let buses = buses
.split(',')
.map(|num| num.parse::<i64>().ok())
.collect_vec();
buses
}
#[aoc(day13, part2)]
fn part2(input: &Vec<Option<i64>>) -> i64 {
let mut min_value = 0;
let mut running_product = 1;
for (index, bus) in input.iter().enumerate() {
if let Some(bus) = bus {
while (min_value + index as i64) % bus != 0 {
min_value += running_product;
}
running_product *= bus;
}
}
min_value
}
#[cfg(test)]
mod tests {
use super::*;
const EX: &str = r"939
7,13,x,x,59,x,31,19";
#[test]
fn part1_example() {
assert_eq!(part1(&parse(EX)), 295);
}
#[test]
fn part2_example() {
assert_eq!(part2(&parse2(EX)), 1068781);
}
}

149
src/day14.rs Normal file
View File

@@ -0,0 +1,149 @@
use std::{collections::HashMap, iter::zip};
use aoc_runner_derive::{aoc, aoc_generator};
use itertools::{Itertools, iproduct};
#[derive(Debug)]
enum ProgramStep {
Mask(String),
Mem(u64, u64)
}
impl From<&str> for ProgramStep {
fn from(value: &str) -> Self {
let (name, value) = value.split(" = ").collect_tuple().unwrap();
return match name {
"mask" => Self::Mask(value.to_string()),
name if name.starts_with("mem") => {
Self::Mem(name[4..name.len() - 1].parse::<u64>().unwrap(), value.parse::<u64>().unwrap())
}
_ => unreachable!()
};
}
}
#[aoc_generator(day14)]
fn parse(input: &str) -> Vec<ProgramStep> {
input.lines().map(ProgramStep::from).collect_vec()
}
fn apply_mask(mask: &str, val: u64) -> u64 {
let mut val = val;
for (i, char) in mask.chars().enumerate() {
match char {
'X' => continue,
'1' => val = val | (2_u64.pow((36 - i - 1).try_into().unwrap())),
'0' => val = val & !(2_u64.pow((36 - i - 1).try_into().unwrap())),
_ => unreachable!()
};
}
val
}
#[aoc(day14, part1)]
fn part1(program: &Vec<ProgramStep>) -> u64 {
let mut mem: HashMap<u64, u64> = HashMap::new();
let mut mask = "";
for step in program {
match step {
ProgramStep::Mask(val) => mask = val,
ProgramStep::Mem(addr, val) => {
let val = apply_mask(mask, *val);
mem.insert(*addr, val);
},
}
}
mem.values().sum()
}
fn permute_masks(mask: &str) -> Vec<String> {
let mut masks: Vec<String> = vec![];
let bits = mask.chars().enumerate().filter(|(_, v)| *v == 'X').map(|(i, _)| i).collect_vec();
let perms = kproduct("21".to_string(), bits.len() as u32);
let mut new_mask = mask.to_string();
for perm in perms {
let bit_vals = zip(bits.clone(), perm.chars()).collect::<HashMap<usize, char>>();
for (k, val) in bit_vals {
new_mask = new_mask[0..k].to_string() + &val.to_string() + &new_mask[k+1..];
}
masks.push(new_mask.clone());
}
masks
}
fn kproduct(seq: String, k: u32) -> Vec<String> {
match k {
0 => vec![],
1 => seq.chars().map(|c| c.to_string()).collect(),
2 => iproduct!(seq.chars(), seq.chars()).map(|(a, b)| format!("{}{}", a, b)).collect(),
_ => iproduct!(kproduct(seq.clone(), k - 1), seq.chars()).map(|(a, b)| format!("{}{}", a, b)).collect(),
}
}
fn get_addrs(addr: u64, mask: &str) -> Vec<u64> {
let mut addrs = vec![];
let masks = permute_masks(mask);
for mask in masks {
let mut addr = addr;
for (i, char) in mask.chars().enumerate() {
match char {
'1' => addr = addr | (2_u64.pow((36 - i - 1).try_into().unwrap())),
'2' => addr = addr & !(2_u64.pow((36 - i - 1).try_into().unwrap())),
'0' => continue,
_ => unreachable!()
}
}
addrs.push(addr)
}
addrs
}
#[aoc(day14, part2)]
fn part2(program: &Vec<ProgramStep>) -> u64 {
let mut mem: HashMap<u64, u64> = HashMap::new();
let mut mask = "";
for step in program {
match step {
ProgramStep::Mask(val) => mask = val,
ProgramStep::Mem(addr, val) => {
let addrs = get_addrs(*addr, mask);
for addr in addrs {
mem.insert(addr, *val);
}
},
}
}
mem.values().sum()
}
#[cfg(test)]
mod tests {
use super::*;
const EX: &str = r"mask = XXXXXXXXXXXXXXXXXXXXXXXXXXXXX1XXXX0X
mem[8] = 11
mem[7] = 101
mem[8] = 0";
#[test]
fn part1_example() {
assert_eq!(part1(&parse(EX)), 165);
}
const EX_2: &str = r"mask = 000000000000000000000000000000X1001X
mem[42] = 100
mask = 00000000000000000000000000000000X0XX
mem[26] = 1";
#[test]
fn part2_example() {
assert_eq!(part2(&parse(EX_2)), 208);
}
}

60
src/day15.rs Normal file
View File

@@ -0,0 +1,60 @@
use std::collections::HashMap;
use aoc_runner_derive::{aoc, aoc_generator};
use itertools::Itertools;
#[aoc_generator(day15)]
fn parse(input: &str) -> Vec<u64> {
input.split(',').map(str::parse::<u64>).map(Result::unwrap).collect_vec()
}
#[aoc(day15, part1)]
fn part1(input: &Vec<u64>) -> u64 {
let mut list = input.iter().take(input.len() - 1).cloned().enumerate().map(|(i,v)| (v,i)).collect::<HashMap<_, _>>();
let mut cur = *input.last().unwrap();
for turn in input.len() - 1..2020 - 1 {
let mut next = 0;
if let Some(last_turn) = list.get_mut(&cur) {
next = (turn - *last_turn) as u64;
}
list.insert(cur, turn);
cur = next;
}
cur
}
#[aoc(day15, part2)]
fn part2(input: &Vec<u64>) -> u64 {
let mut list = input.iter().take(input.len() - 1).cloned().enumerate().map(|(i,v)| (v,i)).collect::<HashMap<_, _>>();
let mut cur = *input.last().unwrap();
for turn in input.len() - 1..30000000 - 1 {
let mut next = 0;
if let Some(last_turn) = list.get_mut(&cur) {
next = (turn - *last_turn) as u64;
}
list.insert(cur, turn);
cur = next;
}
cur
}
#[cfg(test)]
mod tests {
use super::*;
const EX: &str = r"0,3,6";
#[test]
fn part1_example() {
assert_eq!(part1(&parse(EX)), 436);
}
#[test]
fn part2_example() {
assert_eq!(part2(&parse(EX)), 175594);
}
}

145
src/day16.rs Normal file
View File

@@ -0,0 +1,145 @@
use std::{ops::RangeInclusive, collections::HashSet};
use aoc_runner_derive::{aoc, aoc_generator};
use itertools::Itertools;
use regex::Regex;
lazy_static!(
static ref RULE_RE: Regex = Regex::new(r"(.*): (\d+)-(\d+) or (\d+)-(\d+)").unwrap();
);
#[aoc_generator(day16)]
fn parse(input: &str) -> (Vec<(String, RangeInclusive<usize>, RangeInclusive<usize>)>, Vec<usize>, Vec<Vec<usize>>) {
let (rules, my_ticket, tickets) = input
.split("\n\n")
.collect_tuple()
.unwrap();
let rules = RULE_RE
.captures_iter(rules)
.filter_map(|captures| {
Some((
captures.get(1)?.as_str().to_string(),
captures.get(2)?.as_str().parse::<usize>().ok()?..=captures.get(3)?.as_str().parse::<usize>().ok()?,
captures.get(4)?.as_str().parse::<usize>().ok()?..=captures.get(5)?.as_str().parse::<usize>().ok()?,
))
}).collect_vec();
let my_ticket = my_ticket
.lines()
.last()
.unwrap()
.split(',')
.filter_map(|num| num.parse::<usize>().ok())
.collect_vec();
let tickets = tickets
.lines()
.skip(1)
.map(|line| {
line
.split(',')
.filter_map(|num| num.parse::<usize>().ok())
.collect_vec()
}).collect_vec();
(rules, my_ticket, tickets)
}
#[aoc(day16, part1)]
fn part1((rules, _, tickets): &(Vec<(String, RangeInclusive<usize>, RangeInclusive<usize>)>, Vec<usize>, Vec<Vec<usize>>)) -> usize {
let mut invalid = 0;
for ticket in tickets {
for item in ticket {
if !rules.iter().any(|rule| {
rule.1.contains(item) || rule.2.contains(item)
}) {
invalid += item;
}
}
}
invalid
}
#[aoc(day16, part2)]
fn part2((rules, my_ticket, tickets): &(Vec<(String, RangeInclusive<usize>, RangeInclusive<usize>)>, Vec<usize>, Vec<Vec<usize>>)) -> usize {
let mut tickets = tickets.clone();
let mut rem = 0;
for (i, ticket) in tickets.clone().iter().enumerate() {
for item in ticket {
if !rules.iter().any(|rule| {
rule.1.contains(item) || rule.2.contains(item)
}) {
tickets.remove(i - rem);
rem += 1;
}
}
}
let mut possible_rules = get_possible_rules(&tickets, rules);
let mut assigned_rules = [0; 20];
while let Some(i) = possible_rules.iter().position(|s| s.len() == 1) {
let v = *possible_rules[i].iter().next().unwrap();
assigned_rules[i] = v;
for s in &mut possible_rules { s.remove(&v); }
}
assigned_rules.iter()
.enumerate()
.filter(|(_,&rule)| rule < 6)
.map(|(i,_)| my_ticket[i])
.product()
}
fn get_possible_rules(tickets: &Vec<Vec<usize>>, rules: &Vec<(String, RangeInclusive<usize>, RangeInclusive<usize>)>) -> Vec<HashSet<usize>> {
(0..20).map(|i|
(0..20).filter(|&j|
tickets.iter().all(|t| fits_rule(&rules[j], t[i]))
).collect()
).collect()
}
fn fits_rule((_, r1,r2): &(String, RangeInclusive<usize>, RangeInclusive<usize>), v: usize) -> bool {
r1.contains(&v) || r2.contains(&v)
}
#[cfg(test)]
mod tests {
use super::*;
const EX: &str = r"class: 1-3 or 5-7
row: 6-11 or 33-44
seat: 13-40 or 45-50
your ticket:
7,1,14
nearby tickets:
7,3,47
40,4,50
55,2,20
38,6,12";
#[test]
fn part1_example() {
assert_eq!(part1(&parse(EX)), 71);
}
const EX2: &str = r"class: 0-1 or 4-19
row: 0-5 or 8-19
seat: 0-13 or 16-19
your ticket:
11,12,13
nearby tickets:
3,9,18
15,1,5
5,14,9";
#[test]
fn part2_example() {
assert_eq!(part2(&parse(EX2)), 1716);
}
}

119
src/day7.rs Normal file
View File

@@ -0,0 +1,119 @@
use std::collections::{HashMap, VecDeque};
use aoc_runner_derive::{aoc, aoc_generator};
use regex::Regex;
const GOAL: &str = "shiny gold";
lazy_static! {
static ref LINE_RE: Regex = Regex::new(r"(\w+ \w+) bags contain (.*)").unwrap();
static ref ITEM_RE: Regex = Regex::new(r"(\d+) (\w+ \w+) bags?").unwrap();
}
#[aoc_generator(day7)]
fn parse(input: &str) -> HashMap<String, Vec<(usize, String)>> {
let mut bags = HashMap::<String, Vec<(usize, String)>>::new();
for line in input.lines() {
if let Some((item, items)) = LINE_RE
.captures(line.as_ref())
.and_then(|captures| {
Some((captures.get(1)?.as_str(), captures.get(2)?.as_str()))
}) {
bags.insert(
item.to_string(),
ITEM_RE
.captures_iter(items)
.filter_map(|captures| {
Some((
captures.get(1)?.as_str().parse().ok()?,
captures.get(2)?.as_str().to_string(),
))
}).collect()
);
}
}
bags
}
struct Expand<'a> {
bags: &'a HashMap<String, Vec<(usize, String)>>,
queue: VecDeque<(usize, &'a str)>
}
fn expand<'a>(bags: &'a HashMap<String, Vec<(usize, String)>>, bag: &str) -> Expand<'a> {
Expand {
bags,
queue: bags.get(bag).map_or_else(VecDeque::new, |items| {
items.iter()
.map(|(count, item)| (*count, item.as_str()))
.collect()
}),
}
}
impl<'a> Iterator for Expand<'a> {
type Item = (usize, &'a str);
fn next(&mut self) -> Option<Self::Item> {
self.queue.pop_front().map(|(count, item)| {
if let Some(items) = self.bags.get(item) {
for (subcount, subitem) in items {
self.queue.push_back((count * subcount, subitem));
}
}
(count, item)
})
}
}
#[aoc(day7, part1)]
fn part1(bags: &HashMap<String, Vec<(usize, String)>>) -> usize {
bags.keys()
.filter(|key| expand(&bags, key).any(|(_, item)| item == GOAL))
.count()
}
#[aoc(day7, part2)]
fn part2(bags: &HashMap<String, Vec<(usize, String)>>) -> usize {
expand(bags, GOAL).map(|(count, _)| count).sum()
}
#[cfg(test)]
mod tests {
use super::*;
const EX: &str = r"light red bags contain 1 bright white bag, 2 muted yellow bags.
dark orange bags contain 3 bright white bags, 4 muted yellow bags.
bright white bags contain 1 shiny gold bag.
muted yellow bags contain 2 shiny gold bags, 9 faded blue bags.
shiny gold bags contain 1 dark olive bag, 2 vibrant plum bags.
dark olive bags contain 3 faded blue bags, 4 dotted black bags.
vibrant plum bags contain 5 faded blue bags, 6 dotted black bags.
faded blue bags contain no other bags.
dotted black bags contain no other bags.";
const EX_2: &str = r"shiny gold bags contain 2 dark red bags.
dark red bags contain 2 dark orange bags.
dark orange bags contain 2 dark yellow bags.
dark yellow bags contain 2 dark green bags.
dark green bags contain 2 dark blue bags.
dark blue bags contain 2 dark violet bags.
dark violet bags contain no other bags.";
#[test]
fn part1_example() {
assert_eq!(part1(&parse(EX)), 4);
}
#[test]
fn part2_example() {
assert_eq!(part2(&parse(EX)), 32);
}
#[test]
fn part2_example2() {
assert_eq!(part2(&parse(EX_2)), 126);
}
}

180
src/day8.rs Normal file
View File

@@ -0,0 +1,180 @@
use aoc_runner_derive::{aoc, aoc_generator};
use itertools::Itertools;
#[aoc_generator(day8)]
fn parse(input: &str) -> Box<[Instruction]> {
let input: Vec<Instruction> = input.lines()
.map(|line| {
let (instruction, arg) = line.split(' ').collect_tuple().unwrap();
(instruction, arg.parse::<i32>().unwrap()).into()
}).collect_vec();
input.try_into().unwrap()
}
use Instruction::*;
#[derive(Debug, Clone, Copy)]
enum Instruction {
ACC(i32),
JMP(i32),
NOP(i32)
}
impl From<(&str, i32)> for Instruction {
fn from(value: (&str, i32)) -> Self {
match value {
(i, v) if i == "acc" => ACC(v),
(i, v) if i == "jmp" => JMP(v),
(i, v) if i == "nop" => NOP(v),
_ => unreachable!()
}
}
}
#[aoc(day8, part1)]
fn part1(input: &[Instruction]) -> i32 {
let mut visited: Vec<bool> = vec![false; input.len()];
let mut next = 0;
let mut acc = 0;
loop {
let inst = &input[next];
if visited[next] {
break;
}
visited[next] = true;
match inst {
ACC(arg) => acc += arg,
JMP(arg) => {next = (next as i32 + *arg) as usize; continue;},
NOP(_) => {},
}
next += 1;
}
acc
}
#[aoc(day8, part2)]
fn part2(input: &[Instruction]) -> i32 {
let mut visited: Vec<bool> = vec![false; input.len()];
let mut next = 0;
let mut acc = 0;
loop {
let inst = &input[next];
if visited[next] {
break;
}
visited[next] = true;
match inst {
ACC(arg) => acc += arg,
JMP(arg) => {next = (next as i32 + *arg) as usize; continue;},
NOP(_) => {},
}
next += 1;
}
let mut landing_spots = vec![false; input.len() + 1];
let mut i = input.len();
loop {
landing_spots[i] = true;
i -= 1;
if let JMP(x) = input[i] {
if x < 0 {
break;
}
}
}
let start = i;
let swap = if visited[i] {
i
} else {
loop {
i -= 1;
if landing_spots[i] {
continue;
} else if let NOP(x) = input[i] {
if visited[i] && landing_spots[((i as i32) + x) as usize] {
break i;
}
} else if let JMP(x) = input[i] {
if !visited[i]
&& landing_spots[((i as i32) + x) as usize]
&& !landing_spots[i]
{
let mut j = i - 1;
loop {
if matches!(input[j], JMP(_)) {
break;
}
j -= 1;
}
if visited[j] {
break j;
} else {
landing_spots[j + 1..=i].iter_mut().for_each(|a| {
*a = true;
});
i = start;
}
}
}
}
};
let mut program = input.to_vec();
program[swap] = match program[swap] {
ACC(_) => unreachable!(),
JMP(x) => NOP(x),
NOP(x) => JMP(x),
};
let mut next = 0;
let mut acc = 0;
loop {
if next >= program.len() {
break;
}
let inst = &program[next];
match inst {
ACC(arg) => acc += arg,
JMP(arg) => {next = (next as i32 + *arg) as usize; continue;},
NOP(_) => {},
}
next += 1;
}
acc
}
#[cfg(test)]
mod tests {
use super::*;
const EX: &str = r"nop +0
acc +1
jmp +4
acc +3
jmp -3
acc -99
acc +1
jmp -4
acc +6";
#[test]
fn part1_example() {
assert_eq!(part1(&parse(EX)), 5);
}
#[test]
fn part2_example() {
assert_eq!(part2(&parse(EX)), 8);
}
}

73
src/day9.rs Normal file
View File

@@ -0,0 +1,73 @@
use aoc_runner_derive::{aoc, aoc_generator};
use itertools::Itertools;
const PREAMBLE_LEN: usize = 5;
#[aoc_generator(day9)]
fn parse(input: &str) -> Vec<u64> {
input.lines().map(str::parse).map(Result::unwrap).collect_vec()
}
#[aoc(day9, part1)]
fn part1(input: &Vec<u64>) -> u64 {
for i in PREAMBLE_LEN..input.len() {
if let None = input[i - PREAMBLE_LEN..i]
.iter()
.combinations(2)
.find(|x| x.iter().cloned().sum::<u64>() == input[i]) {
return input[i];
}
}
unreachable!()
}
#[aoc(day9, part2)]
fn part2(input: &Vec<u64>) -> u64 {
let to_find = part1(input);
for len in 2..input.len() {
if let Some(nums) = input
.windows(len)
.find(|x| x.iter().cloned().sum::<u64>() == to_find) {
return nums.iter().min().unwrap() + nums.iter().max().unwrap();
}
}
unreachable!()
}
#[cfg(test)]
mod tests {
use super::*;
const EX: &str = r"35
20
15
25
47
40
62
55
65
95
102
117
150
182
127
219
299
277
309
576";
#[test]
fn part1_example() {
assert_eq!(part1(&parse(EX)), 127);
}
#[test]
fn part2_example() {
assert_eq!(part2(&parse(EX)), 62);
}
}

View File

@@ -1,3 +1,12 @@
mod day16;
mod day15;
mod day14;
mod day13;
mod day12;
mod day10;
mod day9;
mod day8;
mod day7;
mod day6;
mod day5;
mod day1;
@@ -8,5 +17,7 @@ extern crate aoc_runner;
#[macro_use]
extern crate aoc_runner_derive;
#[macro_use]
extern crate lazy_static;
aoc_lib!{ year = 2020 }