ishan/aoc2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

day2: optimized

Browse Source
This commit is contained in:
Ishan Jain 2023-12-02 15:30:46 +05:30
parent eccd47a130
commit 77fa1331b2
Signed by: ishan
GPG Key ID: 0506DB2A1CC75C27
2 changed files with 105 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

99
src/day2/1.rs
View File

@ -1,62 +1,49 @@
#![feature(test)] #![feature(test)]
extern crate test; extern crate test;
const INPUTS: [&str; 2] = [include_str!("./sample.txt"), include_str!("./input.txt")]; const INPUTS: [&[u8]; 2] = [
include_bytes!("./sample.txt"),
include_bytes!("./input.txt"),
];
#[derive(Debug, Copy, Clone)] fn process(data: &[u8]) -> u64 {
pub enum Move {
Red(u32),
Green(u32),
Blue(u32),
Unknown,
}
fn process(data: &str) -> u64 {
let mut total = 0; let mut total = 0;
for line in data.lines() { for line in data.split(|&x| x == b'\n') {
if line.is_empty() { if line.is_empty() {
continue; continue;
} }
let (gid, remain) = line.split_once(':').unwrap(); let line = &line[5..];
let (gid, mut line) = read_int(line);
let gid = gid
.split_once(' ')
.map(|(_, x)| x.parse::<u64>().unwrap())
.unwrap();
let moves = remain.split(';').map(|x| {
let mut output = 0u64;
let cubes = x.split(',');
for cube in cubes {
let cube = cube.trim();
let (count, color) = cube.split_once(' ').unwrap();
let count = count.parse::<u64>().unwrap();
match color {
"red" => output |= count << 32,
"green" => output |= count << 16,
"blue" => output |= count,
_ => unreachable!(),
}
}
output
});
let mut possible = true; let mut possible = true;
for mmove in moves {
let red = (mmove & (0xffff << 32)) >> 32;
let green = (mmove & (0xffff << 16)) >> 16;
let blue = mmove & (0xffff);
if red > 12 || green > 13 || blue > 14 { while !line.is_empty() {
let (num, l) = read_int(&line[1..]);
match l {
[b'r', ..] => {
line = &l[3..];
if num > 12 {
possible = false; possible = false;
break;
} }
} }
[b'g', ..] => {
line = &l[5..];
if num > 13 {
possible = false;
}
}
[b'b', ..] => {
line = &l[4..];
if num > 14 {
possible = false;
}
}
[b' ', ..] => line = l,
_ => (),
}
}
if possible { if possible {
total += gid; total += gid;
} }
@ -65,6 +52,28 @@ fn process(data: &str) -> u64 {
total total
} }
fn read_int(line: &[u8]) -> (u64, &[u8]) {
let mut num = 0u64;
let mut multiplier = 10_000u64;
let mut ops_done = 0;
for i in 0..line.len() {
let c = line[i];
if c.is_ascii_digit() {
let c = line[i] - b'0';
ops_done += 1;
num += multiplier * c as u64;
multiplier /= 10;
} else if num != 0 {
return (num / (10u64.pow(5 - ops_done)), &line[i + 1..]);
} else {
break;
}
}
(num / (10u64.pow(5 - ops_done)), line)
}
fn main() { fn main() {
for input in INPUTS.iter() { for input in INPUTS.iter() {
println!("total = {}", process(input)); println!("total = {}", process(input));

83
src/day2/2.rs
View File

@ -1,50 +1,45 @@
#![feature(slice_split_once)]
#![feature(test)] #![feature(test)]
extern crate test; extern crate test;
const INPUTS: [&str; 2] = [include_str!("./sample.txt"), include_str!("./input.txt")]; const INPUTS: [&[u8]; 2] = [
include_bytes!("./sample.txt"),
include_bytes!("./input.txt"),
];
fn process(data: &str) -> u64 { fn process(data: &[u8]) -> u64 {
let mut total = 0; let mut total = 0;
for line in data.lines() { for line in data.split(|&x| x == b'\n') {
if line.is_empty() { if line.is_empty() {
continue; continue;
} }
let (_, remain) = line.split_once(':').unwrap();
let moves = remain.split(';').map(|x| {
let mut output = 0u64;
let cubes = x.split(',');
for cube in cubes {
let cube = cube.trim();
let (count, color) = cube.split_once(' ').unwrap();
let count = count.parse::<u64>().unwrap();
match color {
"red" => output |= count << 32,
"green" => output |= count << 16,
"blue" => output |= count,
_ => unreachable!(),
}
}
output
});
let mut min_red = 0; let mut min_red = 0;
let mut min_green = 0; let mut min_green = 0;
let mut min_blue = 0; let mut min_blue = 0;
for mmove in moves { let line = &line[5..];
let red = (mmove & (0xffff << 32)) >> 32; let (_, mut line) = read_int(line);
let green = (mmove & (0xffff << 16)) >> 16;
let blue = mmove & 0xffff;
min_red = min_red.max(red); while !line.is_empty() {
min_green = min_green.max(green); let (num, l) = read_int(&line[1..]);
min_blue = min_blue.max(blue);
match l {
[b'r', ..] => {
min_red = min_red.max(num);
line = &l[3..];
}
[b'g', ..] => {
min_green = min_green.max(num);
line = &l[5..];
}
[b'b', ..] => {
min_blue = min_blue.max(num);
line = &l[4..];
}
[b' ', ..] => line = l,
_ => (),
}
} }
total += min_red * min_green * min_blue; total += min_red * min_green * min_blue;
@ -53,6 +48,28 @@ fn process(data: &str) -> u64 {
total total
} }
fn read_int(line: &[u8]) -> (u64, &[u8]) {
let mut num = 0u64;
let mut multiplier = 10_000u64;
let mut ops_done = 0;
for i in 0..line.len() {
let c = line[i];
if c.is_ascii_digit() {
let c = line[i] - b'0';
ops_done += 1;
num += multiplier * c as u64;
multiplier /= 10;
} else if num != 0 {
return (num / (10u64.pow(5 - ops_done)), &line[i + 1..]);
} else {
break;
}
}
(num / (10u64.pow(5 - ops_done)), line)
}
fn main() { fn main() {
for input in INPUTS.iter() { for input in INPUTS.iter() {
println!("total = {}", process(input)); println!("total = {}", process(input));