Day 9: Cleaning up garbage

1 files changed, 42 insertions, 0 deletions

diff --git a/src/bin/day_9.rs b/src/bin/day_9.rs
new file mode 100644
index 0000000..e0ca0c0
--- /dev/null
+++ b/src/bin/day_9.rs
@@ -0,0 +1,42 @@
+extern crate advent_of_code_2017;
+use advent_of_code_2017::*;
+
+fn main() {
+    let args = AdventArgs::init();
+
+    let mut cancelled = false;
+    let mut in_garbage = false;
+    
+    let mut depth = 0;
+    let mut total_score = 0;
+    let mut total_garbage = 0;
+    
+    for c in args.input[0].chars() {
+        if cancelled {
+            cancelled = false;
+        } else if c == '!' {
+            cancelled = true;
+        } else if in_garbage {
+            if c == '>' {
+                in_garbage = false;
+            } else {
+                total_garbage += 1;
+            }
+        } else {
+            if c == '<' {
+                in_garbage = true;
+            } else if c == '{' {
+                depth += 1;
+                total_score += depth;
+            } else if c == '}' {
+                depth -= 1;
+            }
+        }
+    }
+
+    if args.part == 1 {
+        println!("Total score is {}", total_score);
+    } else {
+        println!("Total garbage is {}", total_garbage);
+    }
+}