diff --git a/day02/part2.py b/day02/part2.py index 3809a7c..e8cad95 100644 --- a/day02/part2.py +++ b/day02/part2.py @@ -7,8 +7,7 @@ items = map(lambda x: [int(x[0]), int(x[1]), *x[2:]], items) def in_range(lower, upper, char, string): - cnt_range = sum(1 for s in string[lower-1]+string[upper-1] if char == s) - return cnt_range == 1 + return (string[lower-1] == char) != (string[upper-1] == char) count = sum(1 for x in items if in_range(*x))