Update day02.py

This commit is contained in:
kamoshi 2020-12-02 08:47:35 +01:00 committed by GitHub
parent 207c27aea4
commit 4722b81c8f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -2,12 +2,11 @@ def part1():
result = 0
with open("input2.txt") as f:
for line in f:
(policy, char, passcode) = line.split()
(policy, (char,_), passcode) = line.split()
min_p, max_p = policy.split("-")
number_of_char = passcode.count(char[0])
number_of_char = passcode.count(char)
if int(min_p) <= number_of_char <= int(max_p):
result += 1
print(result)
@ -15,9 +14,9 @@ def part2():
result = 0
with open("input2.txt") as f:
for line in f:
(policy, char, passcode) = line.split()
(policy, (char,_), passcode) = line.split()
min_p, max_p = policy.split("-")
if (passcode[int(min_p)-1] == char[0] and passcode[int(max_p)-1] != char[0]) or (passcode[int(min_p)-1] != char[0] and passcode[int(max_p)-1] == char[0]):
if (passcode[int(min_p)-1] == char and passcode[int(max_p)-1] != char) or (passcode[int(min_p)-1] != char and passcode[int(max_p)-1] == char):
result += 1
print(result)