From d828b1a78e98dd29c837d0321876ffe3256a5af0 Mon Sep 17 00:00:00 2001 From: KevinMidboe Date: Sat, 16 Dec 2017 15:15:17 +0100 Subject: [PATCH] Return answer in the wanted format for our tests. --- luke_16/grinchen.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/luke_16/grinchen.py b/luke_16/grinchen.py index 24d5b6e..cbec196 100755 --- a/luke_16/grinchen.py +++ b/luke_16/grinchen.py @@ -16,18 +16,20 @@ def main(): counter = 0 for i, entry in enumerate(entry_list): entry = int(entry)-1 - if entry == 0 and light_state: - counter += 1 - light_state = False - if (counter >= 99): - highest_value.append(i) + if entry == 0: + if light_state: + counter += 1 + light_state = False + if (counter >= 99): + highest_value.append(i) else: if memory[entry] == False and light_state == False: memory[entry] = True light_state = True - - print(min(highest_value)+1) + answer = min(highest_value)+1 + print(answer) + return answer if __name__ == '__main__': main()