2016/12

2016-12-18 13:26:33 +01:00
parent 62bd430c86
commit c8f69b1a50
5 changed files with 324 additions and 0 deletions
--- a/2016/12/Cargo.lock
+++ b/2016/12/Cargo.lock
@@ -0,0 +1,105 @@
+[root]
+name = "12"
+version = "0.1.0"
+dependencies = [
+ "lazy_static 0.1.16 (registry+https://github.com/rust-lang/crates.io-index)",
+ "regex 0.1.80 (registry+https://github.com/rust-lang/crates.io-index)",
+]
+
+[[package]]
+name = "aho-corasick"
+version = "0.5.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+dependencies = [
+ "memchr 0.1.11 (registry+https://github.com/rust-lang/crates.io-index)",
+]
+
+[[package]]
+name = "kernel32-sys"
+version = "0.2.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+dependencies = [
+ "winapi 0.2.8 (registry+https://github.com/rust-lang/crates.io-index)",
+ "winapi-build 0.1.1 (registry+https://github.com/rust-lang/crates.io-index)",
+]
+
+[[package]]
+name = "lazy_static"
+version = "0.1.16"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+
+[[package]]
+name = "libc"
+version = "0.2.18"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+
+[[package]]
+name = "memchr"
+version = "0.1.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+dependencies = [
+ "libc 0.2.18 (registry+https://github.com/rust-lang/crates.io-index)",
+]
+
+[[package]]
+name = "regex"
+version = "0.1.80"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+dependencies = [
+ "aho-corasick 0.5.3 (registry+https://github.com/rust-lang/crates.io-index)",
+ "memchr 0.1.11 (registry+https://github.com/rust-lang/crates.io-index)",
+ "regex-syntax 0.3.9 (registry+https://github.com/rust-lang/crates.io-index)",
+ "thread_local 0.2.7 (registry+https://github.com/rust-lang/crates.io-index)",
+ "utf8-ranges 0.1.3 (registry+https://github.com/rust-lang/crates.io-index)",
+]
+
+[[package]]
+name = "regex-syntax"
+version = "0.3.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+
+[[package]]
+name = "thread-id"
+version = "2.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+dependencies = [
+ "kernel32-sys 0.2.2 (registry+https://github.com/rust-lang/crates.io-index)",
+ "libc 0.2.18 (registry+https://github.com/rust-lang/crates.io-index)",
+]
+
+[[package]]
+name = "thread_local"
+version = "0.2.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+dependencies = [
+ "thread-id 2.0.0 (registry+https://github.com/rust-lang/crates.io-index)",
+]
+
+[[package]]
+name = "utf8-ranges"
+version = "0.1.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+
+[[package]]
+name = "winapi"
+version = "0.2.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+
+[[package]]
+name = "winapi-build"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+
+[metadata]
+"checksum aho-corasick 0.5.3 (registry+https://github.com/rust-lang/crates.io-index)" = "ca972c2ea5f742bfce5687b9aef75506a764f61d37f8f649047846a9686ddb66"
+"checksum kernel32-sys 0.2.2 (registry+https://github.com/rust-lang/crates.io-index)" = "7507624b29483431c0ba2d82aece8ca6cdba9382bff4ddd0f7490560c056098d"
+"checksum lazy_static 0.1.16 (registry+https://github.com/rust-lang/crates.io-index)" = "cf186d1a8aa5f5bee5fd662bc9c1b949e0259e1bcc379d1f006847b0080c7417"
+"checksum libc 0.2.18 (registry+https://github.com/rust-lang/crates.io-index)" = "a51822fc847e7a8101514d1d44e354ba2ffa7d4c194dcab48870740e327cac70"
+"checksum memchr 0.1.11 (registry+https://github.com/rust-lang/crates.io-index)" = "d8b629fb514376c675b98c1421e80b151d3817ac42d7c667717d282761418d20"
+"checksum regex 0.1.80 (registry+https://github.com/rust-lang/crates.io-index)" = "4fd4ace6a8cf7860714a2c2280d6c1f7e6a413486c13298bbc86fd3da019402f"
+"checksum regex-syntax 0.3.9 (registry+https://github.com/rust-lang/crates.io-index)" = "f9ec002c35e86791825ed294b50008eea9ddfc8def4420124fbc6b08db834957"
+"checksum thread-id 2.0.0 (registry+https://github.com/rust-lang/crates.io-index)" = "a9539db560102d1cef46b8b78ce737ff0bb64e7e18d35b2a5688f7d097d0ff03"
+"checksum thread_local 0.2.7 (registry+https://github.com/rust-lang/crates.io-index)" = "8576dbbfcaef9641452d5cf0df9b0e7eeab7694956dd33bb61515fb8f18cfdd5"
+"checksum utf8-ranges 0.1.3 (registry+https://github.com/rust-lang/crates.io-index)" = "a1ca13c08c41c9c3e04224ed9ff80461d97e121589ff27c753a16cb10830ae0f"
+"checksum winapi 0.2.8 (registry+https://github.com/rust-lang/crates.io-index)" = "167dc9d6949a9b857f3451275e911c3f44255842c1f7a76f33c55103a909087a"
+"checksum winapi-build 0.1.1 (registry+https://github.com/rust-lang/crates.io-index)" = "2d315eee3b34aca4797b2da6b13ed88266e6d612562a0c46390af8299fc699bc"
--- a/2016/12/Cargo.toml
+++ b/2016/12/Cargo.toml
@@ -0,0 +1,8 @@
+[package]
+name = "12"
+version = "0.1.0"
+authors = ["logaritmisk <anders.e.olsson@gmail.com>"]
+
+[dependencies]
+lazy_static = "0.1"
+regex = "0.1"
--- a/2016/12/example.txt
+++ b/2016/12/example.txt
@@ -0,0 +1,6 @@
+cpy 41 a
+inc a
+inc a
+dec a
+jnz a 2
+dec a
--- a/2016/12/input.txt
+++ b/2016/12/input.txt
@@ -0,0 +1,23 @@
+cpy 1 a
+cpy 1 b
+cpy 26 d
+jnz c 2
+jnz 1 5
+cpy 7 c
+inc d
+dec c
+jnz c -2
+cpy a c
+inc a
+dec b
+jnz b -2
+cpy c b
+dec d
+jnz d -6
+cpy 17 c
+cpy 18 d
+inc a
+dec d
+jnz d -2
+dec c
+jnz c -5
--- a/2016/12/src/main.rs
+++ b/2016/12/src/main.rs
@@ -0,0 +1,182 @@
+#[macro_use] extern crate lazy_static;
+extern crate regex;
+
+use std::collections::HashMap;
+use std::str::FromStr;
+use std::io::{self, BufRead};
+
+use regex::Regex;
+
+
+type Register = String;
+type Value = i64;
+
+
+#[derive(Debug, PartialEq)]
+pub enum Input {
+    Value(Value),
+    Register(Register)
+}
+
+impl FromStr for Input {
+    type Err = ();
+
+    fn from_str(s: &str) -> Result<Input, Self::Err> {
+        if let Ok(value) = s.parse::<Value>() {
+            return Ok(Input::Value(value));
+        }
+
+        if let Ok(register) = s.parse::<Register>() {
+            return Ok(Input::Register(register));
+        }
+
+        Err(())
+    }
+}
+
+
+#[derive(Debug, PartialEq)]
+pub enum Operation {
+    Cpy(Input, Register),
+    Inc(Register),
+    Dec(Register),
+    Jnz(Input, i64)
+}
+
+impl FromStr for Operation {
+    type Err = ();
+
+    fn from_str(s: &str) -> Result<Operation, Self::Err> {
+        lazy_static! {
+            static ref CPY: Regex = Regex::new(r"^cpy ([a-z]|-?\d+) ([a-z])$").unwrap();
+            static ref INC: Regex = Regex::new(r"^inc ([a-z])$").unwrap();
+            static ref DEC: Regex = Regex::new(r"^dec ([a-z])$").unwrap();
+            static ref JNZ: Regex = Regex::new(r"^jnz ([a-z]|-?\d+) (-?\d+)$").unwrap();
+        }
+
+        if let Some(caps) = CPY.captures(s) {
+            let input = caps.at(1).unwrap().parse::<Input>().unwrap();
+            let register = caps.at(2).unwrap().parse::<Register>().unwrap();
+
+            return Ok(Operation::Cpy(input, register));
+        }
+
+        if let Some(caps) = INC.captures(s) {
+            let register = caps.at(1).unwrap().parse::<Register>().unwrap();
+
+            return Ok(Operation::Inc(register));
+        }
+
+        if let Some(caps) = DEC.captures(s) {
+            let register = caps.at(1).unwrap().parse::<Register>().unwrap();
+
+            return Ok(Operation::Dec(register));
+        }
+
+        if let Some(caps) = JNZ.captures(s) {
+            let input = caps.at(1).unwrap().parse::<Input>().unwrap();
+            let delta = caps.at(2).unwrap().parse::<i64>().unwrap();
+
+            return Ok(Operation::Jnz(input, delta));
+        }
+
+
+        Err(())
+    }
+}
+
+
+#[derive(Debug)]
+pub struct CPU {
+    pc: usize,
+    reg: HashMap<Register, Value>,
+}
+
+impl CPU {
+    pub fn new() -> Self {
+        CPU {
+            pc: 0,
+            reg: HashMap::new()
+        }
+    }
+
+    pub fn execute(&mut self, input: &Vec<Operation>) {
+        loop {
+            if self.pc >= input.len() {
+                break;
+            }
+
+            match &input[self.pc] {
+                &Operation::Cpy(ref value, ref register) => {
+                    let input = match value {
+                        &Input::Value(value) => {
+                            value
+                        }
+                        &Input::Register(ref register) => {
+                            *self.reg.entry(register.clone()).or_insert(0)
+                        }
+                    };
+
+                    *self.reg.entry(register.clone()).or_insert(0) = input;
+
+                    self.pc += 1;
+                }
+                &Operation::Inc(ref register) => {
+                    *self.reg.entry(register.clone()).or_insert(0) += 1;
+
+                    self.pc += 1;
+                }
+                &Operation::Dec(ref register) => {
+                    *self.reg.entry(register.clone()).or_insert(0) -= 1;
+
+                    self.pc += 1;
+                }
+                &Operation::Jnz(ref value, delta) => {
+                    let input = match value {
+                        &Input::Value(value) => {
+                            value
+                        }
+                        &Input::Register(ref register) => {
+                            *self.reg.entry(register.clone()).or_insert(0)
+                        }
+                    };
+
+                    if input != 0 {
+                        self.pc = ((self.pc as i64) + delta) as usize;
+                    } else {
+                        self.pc += 1;
+                    }
+                }
+            }
+        }
+    }
+}
+
+
+fn main() {
+    let stdin = io::stdin();
+
+    let operations = stdin.lock().lines()
+        .filter_map(|x| x.ok())
+        .filter_map(|line| line.parse::<Operation>().ok())
+        .collect::<Vec<_>>();
+
+    let mut cpu = CPU::new();
+
+    cpu.reg.entry("c".to_owned()).or_insert(1);
+
+    cpu.execute(&operations);
+
+    println!("{:#?}", cpu);
+}
+
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn parse_jnz() {
+        assert_eq!(Ok(Operation::Jnz(Input::Register("c".to_owned()), -2)), "jnz c -2".parse::<Operation>());
+    }
+}