kamov/advent-of-code
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Update day12.py

Browse source
This commit is contained in:
kamoshi 2020-12-12 13:34:55 +01:00 committed by GitHub
parent 5e6add69a8
commit 1c9fc34d27
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 48 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
2020/Python/day12.py
View file

@ -57,16 +57,60 @@ def apply_command(curr_pos: Tuple[int, int], curr_dir: Direction, command: Tuple
return (curr_x + _number, curr_y), curr_dir
def solve_p1(data: list[Tuple[str, int]]):
def solve_p1(data: list[Tuple[str, int]]) -> int:
current_direction = Direction.EAST
current_position = 0, 0
for command in data:
new_position, new_direction = apply_command(current_position, current_direction, command)
current_direction = new_direction
current_position = new_position
current_position, current_direction = apply_command(current_position, current_direction, command)
final_x, final_y = current_position
return abs(final_x)+abs(final_y)
def apply_command2(curr_pos: Tuple[int, int], curr_wp: Tuple[int, int], command: Tuple[str, int]) -> Tuple[Tuple[int, int], Tuple[int, int]]:
curr_x, curr_y = curr_pos
wp_x, wp_y = curr_wp
_char, _number = command
def find_rotated_wp(_curr_wp: Tuple[int, int], how: str, degrees: int) -> (int, int):
degrees = degrees % 360
_wp_x, _wp_y = _curr_wp
if how == 'R':
degrees = 360 - degrees
if degrees == 0:
return _wp_x, _wp_y
elif degrees == 90:
return -_wp_y, _wp_x
elif degrees == 180:
return -_wp_x, -_wp_y
elif degrees == 270:
return _wp_y, -_wp_x
if _char == 'N':
return curr_pos, (wp_x, wp_y + _number)
elif _char == 'S':
return curr_pos, (wp_x, wp_y - _number)
elif _char == 'E':
return curr_pos, (wp_x + _number, wp_y)
elif _char == 'W':
return curr_pos, (wp_x - _number, wp_y)
elif _char in "LR":
return curr_pos, find_rotated_wp(curr_wp, _char, _number)
elif _char == 'F':
return (curr_x + wp_x * _number, curr_y + wp_y * _number), (wp_x, wp_y)
def solve_p2(data: list[Tuple[str, int]]) -> int:
current_waypoint = (10, 1)
current_position = (0, 0)
for command in data:
current_position, current_waypoint = apply_command2(current_position, current_waypoint, command)
final_x, final_y = current_position
return abs(final_x)+abs(final_y)
DATA = parse_data()
print(solve_p1(DATA))
print(solve_p2(DATA))