diff options
| author | Botond Hende <nettingman@gmail.com> | 2024-12-07 11:08:24 +0100 |
|---|---|---|
| committer | Botond Hende <nettingman@gmail.com> | 2024-12-07 11:08:24 +0100 |
| commit | 5afb1de803675e818b76725e2213b3480f6c3779 (patch) | |
| tree | 6f32a0a327b818ae994003059a5d1864bd83eb5c /2023 | |
| parent | a6149004a4e7627fc3dc0a9ca75d03d7b85cc6a0 (diff) | |
fixed 2023 day3 solve2
Diffstat (limited to '2023')
| -rw-r--r-- | 2023/day3/solve2.py | 104 |
1 files changed, 104 insertions, 0 deletions
diff --git a/2023/day3/solve2.py b/2023/day3/solve2.py index e69de29..5b23f13 100644 --- a/2023/day3/solve2.py +++ b/2023/day3/solve2.py @@ -0,0 +1,104 @@ +class Gear:
+ def __init__(self, line_idx, column_idx):
+ self.line_idx = line_idx
+ self.column_idx = column_idx
+ self.connecting_parts = []
+
+
+def add_part_to_gear(line_idx, column_idx, part_number_line_idx, part_number_column_idx, gears):
+ for gear in gears:
+ if gear.line_idx == line_idx and gear.column_idx == column_idx:
+ gear.connecting_parts.append((part_number_line_idx, part_number_column_idx))
+ break
+ else:
+ new_gear = Gear(line_idx, column_idx)
+ new_gear.connecting_parts.append((part_number_line_idx, part_number_column_idx))
+ gears.append(new_gear)
+
+
+def check_if_part_number(line_index, column_index, lines, gears):
+ ret_val = False
+
+ current_line = lines[line_index]
+ if column_index > 0 and current_line[column_index - 1] != ".":
+ if current_line[column_index - 1] == "*":
+ add_part_to_gear(line_index, column_index - 1, line_index, column_index, gears)
+ ret_val = True
+
+ max_column_index = column_index
+ for index in range(column_index + 1, len(current_line)):
+ if not current_line[index].isdigit():
+ break
+ max_column_index = index
+
+ if max_column_index + 1 < len(current_line) and current_line[max_column_index + 1] != ".":
+ if current_line[max_column_index + 1] == "*":
+ add_part_to_gear(line_index, column_index + 1, line_index, column_index, gears)
+ ret_val = True
+
+ min_other_column_index = max(column_index - 1, 0)
+ max_other_column_index = min(max_column_index + 1, len(current_line) - 1)
+
+ if line_index > 0:
+ other_line = lines[line_index - 1]
+ for current_column_index in range(min_other_column_index, max_other_column_index + 1):
+ other_char = other_line[current_column_index]
+ if not other_char.isdigit() and other_char != ".":
+ if other_char == "*":
+ add_part_to_gear(line_index - 1, current_column_index, line_index, column_index, gears)
+ ret_val = True
+
+ if line_index + 1 < len(lines):
+ other_line = lines[line_index + 1]
+ for current_column_index in range(min_other_column_index, max_other_column_index + 1):
+ other_char = other_line[current_column_index]
+ if not other_char.isdigit() and other_char != ".":
+ if other_char == "*":
+ add_part_to_gear(line_index + 1, current_column_index, line_index, column_index, gears)
+ ret_val = True
+
+ return ret_val
+
+
+def main():
+ lines = []
+ positions = []
+ gears = []
+ with open("input", "r") as f:
+ for line in f:
+ lines.append(line.strip())
+
+ part_numbers = []
+
+ parsed_number = None
+ is_part_number_parsed = False
+ for line_index, line in enumerate(lines):
+ for column_index, char in enumerate(line):
+ if not char.isdigit():
+ if is_part_number_parsed:
+ part_numbers.append(int(parsed_number))
+
+ parsed_number = None
+ is_part_number_parsed = False
+ elif parsed_number is not None:
+ parsed_number += char
+ else:
+ parsed_number = char
+ is_part_number_parsed = check_if_part_number(line_index, column_index, lines, gears)
+ if is_part_number_parsed:
+ positions.append((line_index, column_index))
+
+ if is_part_number_parsed:
+ part_numbers.append(int(parsed_number))
+ parsed_number = None
+ is_part_number_parsed = False
+
+ gears = [gear for gear in gears if len(gear.connecting_parts) == 2]
+ ratios = []
+ for gear in gears:
+ ratios.append(part_numbers[positions.index(gear.connecting_parts[0])] * part_numbers[positions.index(gear.connecting_parts[1])])
+ print(sum(ratios))
+
+
+if __name__ == '__main__':
+ main()
|