2024(5): only add the updated ones

2024/5/puzzle2.py

@@ -47,23 +47,18 @@ def fix_update(update, rules):
 if __name__ == "__main__":
 	rules, updates = load_input()
 
-	indices_legal = []
 	middle_sum = 0
 	# Let's assume numbers can only occur once (although the checker does not
 	# make this assumption), and every update CAN be fixed.
 	for idx, update in enumerate(updates):
 		print(f"{idx:5d}/{len(updates)}:", update)
-		fix_update(update, rules)
-		updates[idx] = update
 		if is_update_legal(update, rules):
 			if verbose:
 				print(update, " is legal")
-			indices_legal.append(idx)
-			middle_sum += middle_page_num(update)
 		else:
-			if True or verbose:		
-				print(update, " is illegal")
-
 			if verbose:
-		print(indices_legal)
+				print(update, " is illegal")
+			fix_update(update, rules)
+			middle_sum += middle_page_num(update)
+
 	print(f"The sum of middle page numbers of legal updates is {middle_sum}.")