Solve 2015/05

2015/05/solve.py

@@ -0,0 +1,54 @@
+import sys
+
+def load_words(filename):
+	words = []
+	with open(filename, "r") as f:
+		for line in f:
+			words.append(line[:-1])
+	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 main(filename):
+	words = load_words(filename)
+	print(f"Solutions for {filename}")
+	amount = count(words, is_nice)
+	print(f"Part 1: {amount}")
+	amount_2 = count(words, is_nice_2)
+	print(f"Part 2: {amount_2}")
+
+if __name__ == "__main__":
+	for filename in sys.argv[1:]:
+		main(filename)