Add day 4 solution

2021-12-04 13:09:22 +01:00
parent 93b7bbedca
commit 8e535ef161
2 changed files with 737 additions and 0 deletions
--- a/day-04/day-04.py
+++ b/day-04/day-04.py
@@ -0,0 +1,136 @@
+#!/usr/bin/python3
+
+def read_boards(input):
+    boards = list()
+    board = list()
+    row = 0
+    for line in input:
+        if not line.rstrip():
+            boards.append(board)
+            board = list()
+            row = 0
+            continue
+        col = 0
+        for number in line.split():
+            field = dict()
+            field['number'] = int(number)
+            field['marked'] = False
+            field['row'] = row
+            field['col'] = col
+            col += 1
+            board.append(field)
+        row += 1
+    return boards
+
+
+def mark_number(boards, number):
+    for board in boards:
+        for field in board:
+            if field['number'] == number:
+                field['marked'] = True
+    return boards
+
+
+def check_board(board, num_rows, num_cols):
+    for row in range(num_rows):
+        win = True
+        for field in board:
+            if not field['marked']:
+                if field['row'] == row:
+                    win = False
+        if win:
+            return True
+    for col in range(num_cols):
+        win = True
+        for field in board:
+            if not field['marked']:
+                if field['col'] == col:
+                    win = False
+        if win:
+            return True
+    return False
+
+
+def get_board_sum(board):
+    sum = 0
+    for field in board:
+        if not field['marked']:
+            sum += field['number']
+    return sum
+
+
+def check_boards(boards, num_rows, num_cols):
+    for board in boards:
+        for row in range(num_rows):
+            win = True
+            sum = 0
+            for field in board:
+                if not field['marked']:
+                    if field['row'] == row:
+                        win = False
+                    sum += field['number']
+            if win:
+                return sum
+        for col in range(num_cols):
+            win = True
+            sum = 0
+            for field in board:
+                if not field['marked']:
+                    if field['col'] == col:
+                        win = False
+                    sum += field['number']
+            if win:
+                return sum
+    return 0
+
+
+def part_1(input):
+    result = 0
+    drawn = input[0].split(',')
+    input.remove(input[0])
+    input.remove(input[0])
+    boards = read_boards(input)
+    for number in drawn:
+        number = int(number)
+        end = False
+        boards = mark_number(boards, number)
+        for board in boards:
+            if check_board(board, 5, 5):
+                end = True
+                result = get_board_sum(board) * number
+                break
+        if end:
+            break
+    print("Part 1 result:", result)
+
+
+def part_2(input):
+    result = 0
+    drawn = input[0].split(',')
+    input.remove(input[0])
+    input.remove(input[0])
+    boards = read_boards(input)
+    last_board = list()
+    for number in drawn:
+        number = int(number)
+        boards = mark_number(boards, number)
+        num_complete = 0
+        for board in boards:
+            if check_board(board, 5, 5):
+                num_complete += 1
+        if len(boards) - num_complete == 1:
+            for board in boards:
+                if not check_board(board, 5, 5):
+                    last_board = board
+        elif last_board and len(boards) == num_complete:
+            result = get_board_sum(last_board) * number
+            break
+    print("Part 2 result:", result)
+
+
+input = list()
+with open('input.txt') as fp:
+    input = fp.readlines()
+
+part_1(input.copy())
+part_2(input.copy())