aoc2015/d05/run.py

18 lines
536 B
Python

import sys
def isnice1(l):
if any(a+b in ['ab','cd','pq','xy'] for (a,b) in zip(l,l[1:])):
return False
vowels = sum(c in 'aeiou' for c in l)
inarow = any(a==b for (a,b) in zip(l,l[1:]))
return vowels > 2 and inarow
def isnice2(l):
xyxy = any(l[i:i+2] == l[j:j+2] for i in range(len(l)-1) for j in range(i+2,len(l)-1))
gfg = any(a == b for a,b in zip(l[0:],l[2:]))
return xyxy and gfg
isnice = isnice2 if sys.argv[1] == "2" else isnice1
print(sum(isnice(l[:-1]) for l in sys.stdin.readlines()))