From 7455d902f18b0daba0589981123a1d5e7fe4cab9 Mon Sep 17 00:00:00 2001
From: Bert Peters <bert.ljpeters@gmail.com>
Date: Tue, 6 Dec 2016 12:21:58 +0100
Subject: [PATCH] Clean up day 06.

---
 2016/day-06/src/main.rs | 13 +++----------
 1 file changed, 3 insertions(+), 10 deletions(-)

diff --git a/2016/day-06/src/main.rs b/2016/day-06/src/main.rs
index 5ff6f6a..7dc1ee6 100644
--- a/2016/day-06/src/main.rs
+++ b/2016/day-06/src/main.rs
@@ -29,8 +29,6 @@ fn least_frequent(counts: &HashMap<char, i32>) -> char
 }
 
 fn main() {
-    println!("Hello, world!");
-
     let args: Vec<String> = env::args().collect();
     let f = File::open(&args[1]).expect("Could not open file");
     let reader = BufReader::new(f);
@@ -39,17 +37,12 @@ fn main() {
 
     for line in reader.lines() {
         for (i, c) in line.unwrap().trim().chars().enumerate() {
-            let pos = i as usize;
-            if pos >= counts.len() {
+            if i >= counts.len() {
                 counts.push(HashMap::new());
             }
 
-            if ! counts[pos].contains_key(&c) {
-                counts[pos].insert(c, 1);
-            } else {
-                let cur = counts[pos][&c];
-                counts[pos].insert(c, cur + 1);
-            }
+            let cur = counts[i].entry(c).or_insert(0);
+            *cur += 1;
         }
     }