Return answer in the wanted format for our tests.

This commit is contained in:
2017-12-16 15:15:17 +01:00
parent 06c422f2dd
commit d828b1a78e

View File

@@ -16,7 +16,8 @@ def main():
counter = 0 counter = 0
for i, entry in enumerate(entry_list): for i, entry in enumerate(entry_list):
entry = int(entry)-1 entry = int(entry)-1
if entry == 0 and light_state: if entry == 0:
if light_state:
counter += 1 counter += 1
light_state = False light_state = False
if (counter >= 99): if (counter >= 99):
@@ -26,8 +27,9 @@ def main():
memory[entry] = True memory[entry] = True
light_state = True light_state = True
answer = min(highest_value)+1
print(min(highest_value)+1) print(answer)
return answer
if __name__ == '__main__': if __name__ == '__main__':
main() main()