bert/adventofcode
1
0
Fork 0
mirror of https://github.com/bertptrs/adventofcode.git synced 2025-12-25 21:00:31 +01:00
Code Issues Projects Releases Wiki Activity

Clean up solution to day 1.

Browse Source
This commit is contained in:
Bert Peters
2016-12-02 11:19:00 +01:00
parent 9f0b9b4ef8
commit 6937fec0fd
1 changed files with 22 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
2016/day-01/solution.rs
View File

@@ -6,12 +6,21 @@ use std::io::prelude::*;
use std::path::Path;
use std::str;
fn dist(pos: (i32, i32)) -> i32 {
fn dist(pos: (i32, i32)) -> i32
{
let (x, y) = pos;
return x.abs() + y.abs()
}
fn walk(pos: (i32, i32), dir: i32) -> (i32, i32)
{
let backwards = if dir & 2 == 2 { -1 } else { 1 };
let (x, y) = pos;
(x + backwards * (1 - (dir & 1)), y + backwards * (dir & 1))
}
fn main() {
let args: Vec<String> = env::args().collect();
let path = Path::new(&args[1]);
@@ -33,13 +42,12 @@ fn main() {
Ok(_) => {},
};
let mut posx = 0;
let mut posy = 0;
let mut pos = (0, 0);
let mut dir = 0;
let mut found = false;
let mut positions = HashSet::new();
positions.insert((posx, posy));
positions.insert(pos);
for instruction in content.split(", ") {
let turn = &instruction[..1];
@@ -51,36 +59,20 @@ fn main() {
if turn == "R" {
dir += 1;
} else {
dir -= 1;
dir += 3;
}
dir = (dir + 4) % 4;
dir %= 4;
let backwards = if dir & 2 == 2 { -1 } else { 1 };
if dir & 1 == 1 {
// Move in y direction
for y in 1..(steps + 1) {
if positions.contains(&(posx, posy + y * backwards)) && !found {
found = true;
println!("Arrived at the same position, dist {}", dist((posx, posy + y * backwards)));
} else {
positions.insert((posx, posy + y * backwards));
}
for _ in 0..steps {
pos = walk(pos, dir);
if !found && positions.contains(&pos) {
println!("Visited twice at dist {}", dist(pos)) ;
found = true;
} else {
positions.insert(pos);
}
posy += steps * backwards;
} else {
// Move in x direction
for x in 1..(steps + 1) {
if positions.contains(&(posx + x * backwards, posy)) && !found {
println!("Arrived at the same position, dist {}", dist((posx + x * backwards, posy)));
found = true;
} else {
positions.insert((posx + x * backwards, posy));
}
}
posx += steps * backwards;
}
}
println!("Total distance is {}", dist((posx, posy)))
println!("Total distance is {}", dist(pos));
}