[py] Port 2015_05

py/aoc/y2015/d05.py

@@ -0,0 +1,56 @@
+def load_words(inputstr):
+    words = []
+    for line in inputstr.splitlines():
+        words.append(line)
+    return words
+
+
+# PART 1
+
+
+def is_vowel(char):
+    return char in "aeiou"
+
+
+def has_double_letter(word):
+    # This would look nicer im haskell:
+    return count(zip(word, word[1:]), lambda x: x[0] == x[1]) > 0
+
+
+def is_nice(word):
+    if "ab" in word or "cd" in word or "pq" in word or "xy" in word:
+        return False
+    if len(list(filter(is_vowel, word))) < 3:
+        return False
+    return has_double_letter(word)
+
+
+def count(what, function):
+    # This would also look nicer in haskell
+    return len(list(filter(function, what)))
+
+
+# PART 2
+
+
+def has_pair(word):
+    for i in range(len(word)):
+        if word[i : i + 2] in word[i + 2 :]:
+            return True
+    return False
+
+
+def has_repeat_letter(word):
+    return count(zip(word, word[2:]), lambda x: x[0] == x[1]) > 0
+
+
+def is_nice_2(word):
+    return has_pair(word) and has_repeat_letter(word)
+
+
+def solve(inputstr):
+    words = load_words(inputstr)
+    amount = count(words, is_nice)
+    print(f"Part 1: {amount}")
+    amount_2 = count(words, is_nice_2)
+    print(f"Part 2: {amount_2}")