diff --git a/d03/run1.py b/d03/run1.py index f36cf16..2cd366a 100644 --- a/d03/run1.py +++ b/d03/run1.py @@ -6,7 +6,7 @@ def if_symbol_adj(l, n, i): for m in [n-1, n, n+1]: if m < 0 or m > len(l)-1: continue for j in [i-1, i, i+1]: - if j < 0 or j > len(l[m]): continue + if j < 0 or j > len(l[m])-1: continue if l[m][j] != "." and l[m][j].isdigit() is False: return True else: @@ -23,10 +23,11 @@ def main(S = 0): if myList[n][i].isdigit(): nb = "".join([nb, myList[n][i]]) if_sum = if_sum or if_symbol_adj(myList, n, i) - else: - # soit pas encore un nombre soit la fin - print(int(nb), if_sum) - S += (int(nb) if nb != "" else 0) + + if not myList[n][i].isdigit() or i == len(myList[n]) -1: + # soit pas encore un nombre soit la fin d'un nombre, soit la fin d'une ligne + #if nb!="": print(nb, if_sum) + S += (int(nb) if nb != "" and if_sum else 0) if_sum = False nb = ""