From 681d0ddfa960e552c81683ac5f44bcd89bd0a71e Mon Sep 17 00:00:00 2001
From: Luke Hubmayer-Werner <ld.hubmayer-werner@gmail.com>
Date: Wed, 5 Dec 2018 11:27:06 +1030
Subject: [PATCH] golfed 2018 day4 solution

---
 2018/day4-cleaned.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 2018/day4-cleaned.py

diff --git a/2018/day4-cleaned.py b/2018/day4-cleaned.py
new file mode 100644
index 0000000..e5c9c9f
--- /dev/null
+++ b/2018/day4-cleaned.py
@@ -0,0 +1,26 @@
+with open('day4-input', 'r') as file:
+  data = sorted([l.strip('\n') for l in file])
+import numpy as np
+
+guards = {}
+guard = None
+guard_sleep = None
+
+for line in data:
+  _, time, *event = line.split()
+  minute = int(time[-3:-1])
+  if event[0] == 'Guard':
+    if guard_sleep is not None:
+      raise Exception("Last guard didn't wake up?!")
+    guard = int(event[1][1:])
+    if guard not in guards:
+      guards[guard] = np.zeros((60))
+  elif event[0] == 'wakes':
+    guards[guard][guard_sleep:minute] += 1
+    guard_sleep = None
+  else:  # falls asleep
+    guard_sleep = minute
+
+for func in (np.sum, np.max):  # Part 1, Part 2. Pattern shamelessly copied from Peter Tseng's Ruby solution.
+  guard, minutes = max(guards.items(), key=lambda x: func(x[1]))
+  print(guard * minutes.argmax())