백준) 1063 - 킹 [파이썬 3]

문제)

 8*8크기의 체스판에 왕이 하나 있다. 킹의 현재 위치가 주어진다. 체스판에서 말의 위치는 다음과 같이 주어진다. 알파벳 하나와 숫자 하나로 이루어져 있는데, 알파벳은 열을 상징하고, 숫자는 행을 상징한다. 열은 가장 왼쪽 열이 A이고, 가장 오른쪽 열이 H까지 이고, 행은 가장 아래가 1이고 가장 위가 8이다. 예를 들어, 왼쪽 아래 코너는 A1이고, 그 오른쪽 칸은 B1이다.

 킹은 다음과 같이 움직일 수 있다.

• R : 한 칸 오른쪽으로
• L : 한 칸 왼쪽으로
• B : 한 칸 아래로
• T : 한 칸 위로
• RT : 오른쪽 위 대각선으로
• LT : 왼쪽 위 대각선으로
• RB : 오른쪽 아래 대각선으로
• LB : 왼쪽 아래 대각선으로

 체스판에는 돌이 하나 있는데, 돌과 같은 곳으로 이동할 때는, 돌을 킹이 움직인 방향과 같은 방향으로 한 칸 이동시킨다. 아래 그림을 참고하자.

 입력으로 킹이 어떻게 움직여야 하는지 주어진다. 입력으로 주어진 대로 움직여서 킹이나 돌이 체스판 밖으로 나갈 경우에는 그 이동은 건너 뛰고 다음 이동을 한다.

 킹과 돌의 마지막 위치를 구하는 프로그램을 작성하시오.

 

입력)

 첫째 줄에 킹의 위치, 돌의 위치, 움직이는 횟수 N이 주어진다. 둘째 줄부터 N개의 줄에는 킹이 어떻게 움직여야 하는지 주어진다. N은 50보다 작거나 같은 자연수이고, 움직이는 정보는 위에 쓰여 있는 8가지 중 하나이다.

 

출력)

 첫째 줄에 킹의 마지막 위치, 둘째 줄에 돌의 마지막 위치를 출력한다.

---

from sys import stdin; input=stdin.readline

def check_out(arr):
    # 체스판을 나가면 True, 아니면 False
    result = True if arr[0] < 1 or arr[0] > 8 or arr[1] < 1 or arr[1] > 8 else False
    return int(result)

def moving(command, arr):
    # Arr[0] : 행 A~H, Arr[1]: 열 1~8
    # RL : 행 이동, TB : 열 이동
    result = arr.copy()
    if command == 'R': result[0] += 1
    elif command == 'L': result[0] -= 1
    elif command == 'T': result[1] += 1
    elif command == 'B': result[1] -= 1
    elif command == 'RT': result[0] += 1; result[1] += 1
    elif command == 'LT': result[0] -= 1; result[1] += 1
    elif command == 'RB': result[0] += 1; result[1] -= 1
    elif command == 'LB': result[0] -= 1; result[1] -= 1

    return result

king, stone, move_count = input().split()
COORD = '_ABCDEFGH'
king_arr = [COORD.index(king[0]), int(king[1])]
stone_arr = [COORD.index(stone[0]), int(stone[1])]

for _ in range(int(move_count)):
    command = input().strip()
    new_king_arr = moving(command, king_arr.copy())

    new_stone_arr = stone_arr.copy()
    if new_king_arr == new_stone_arr:
        new_stone_arr = moving(command, stone_arr)
    
    is_out = check_out(new_king_arr) + check_out(new_stone_arr)

    if not is_out: # 이동 후 좌표들이 체스판을 벗어나지 않았을 경우
        king_arr = new_king_arr
        stone_arr = new_stone_arr

print(COORD[king_arr[0]]+str(king_arr[1]))
print(COORD[stone_arr[0]]+str(stone_arr[1]))

풀이 : 단순 구현이므로 별도의 풀이는 필요없음