You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.1 KiB
44 lines
1.1 KiB
import os
|
|
|
|
dir = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
with open(os.path.join(dir, './input.txt'), 'r') as input_file:
|
|
|
|
lines = input_file.readlines()
|
|
|
|
ones = 0
|
|
fours = 0
|
|
sevens = 0
|
|
eights = 0
|
|
|
|
test = False
|
|
|
|
for line in lines:
|
|
input_segments = ""
|
|
output_segments = ""
|
|
|
|
if test == True:
|
|
if '|' in line:
|
|
input_segments = line.strip()
|
|
else:
|
|
output_segments = line.strip()
|
|
else:
|
|
parts = line.split('|')
|
|
input_segments = parts[0].strip()
|
|
output_segments = parts[1].strip()
|
|
|
|
if output_segments == "":
|
|
continue
|
|
|
|
output_parts = output_segments.split()
|
|
for output in output_parts:
|
|
num_signals = len(output)
|
|
if num_signals == 2:
|
|
ones = ones + 1
|
|
elif num_signals == 3:
|
|
sevens = sevens + 1
|
|
elif num_signals == 4:
|
|
fours = fours + 1
|
|
elif num_signals == 7:
|
|
eights = eights + 1
|
|
print(ones+fours+sevens+eights)
|