fixed last column logic

sullivancolin · Dec 10, 2022 · 7a07edd · 7a07edd
1 parent fdddfad
commit 7a07edd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/advent_of_code/day10/part2.py b/src/advent_of_code/day10/part2.py
@@ -30,8 +30,11 @@
 
 
 def get_pixel_val(sprite_location: np.ndarray, curr_cycle: int) -> str:
+    orig_cycle = curr_cycle
     if curr_cycle > 40:
         curr_cycle = curr_cycle % 40
+        if curr_cycle == 0:
+            curr_cycle = 40
     if any(sprite_location == curr_cycle - 1):
         pixel = "#"
     else:
@@ -59,7 +62,7 @@ def print_screen(input_str: str) -> str:
             pixel_val = get_pixel_val(sprite_location, curr_cycle)
             pixels.append(pixel_val)
             sprite_location += instruction
-    return "".join(pixels)
+    return "".join(pixels).strip()
 
 
 if __name__ == "__main__":