[이취코테] 왕실의 나이트

https://youtu.be/2zjoKjt97vQ

이 문제의 해설 영상이다.

문제

체스판과 같은 8 x 8 좌표 평면에서 나이트는 특정한 위치에서 다음과 같은 2가지 경우로 이동할 수 있다.

1. 수평으로 두 칸 이동한 뒤에 수직으로 한 칸 이동하기
2. 수직으로 두 칸 이동한 뒤에 수평으로 한 칸 이동하기

 8 x 8 좌표 평면 상에서 나이트의 위치가 주어졌을 때 나이트가 이동할 수 있는 경우의 수를 출력하는 프로그램을 작성하시오. (단, 좌표평면 밖으로는 나갈 수 없다.) 이때 행 위치는 1부터 8로 표현하며, 열 위치는 a부터 h로 표현한다.

 

입력 조건:

• 첫째 줄에 8 x 8 좌표 평면 상에서 현재 나이트가 위치한 곳의 좌표를 나타내는 두 문자로 된 문자열이 입력된다.
• 입력 문자는 a1처럼 열과 행으로 이루어진다.

 

출력 조건: 첫째 줄에 나이트가 이동할 수 있는 경우의 수를 출력한다.

 

나의 코드

position = input()

dx = [-1, 1, -1, 1, 2, 2, -2, -2]
dy = [2, 2, -2, -2, -1, 1, -1, 1]

cnt = 0

for i in range(8):
    x = int(position[1])
    y = ord(position[0]) - ord('a') + 1
    
    x += dx[i]
    y += dy[i]
    
    if 1 <= x <= 8 and 1 <= y <= 8:
        cnt += 1

print(cnt)

이 문제는 책에서 앞서 풀어본 예제 4 - 1 '상하좌우' 문제와 유사하다. 풀이 방법은 이동할 수 있는 방향벡터를 리스트로 저장해놓는 것이다. 각각의 방향벡터로 이동한 후의 x, y 좌표를 검사하여 좌표 평면을 벗어나지 않았다면 cnt ++ 해준다. 

 

이 문제를 풀 때 약간 고민했던 부분은 y좌표가 알파벳으로 주어져서 그것을 숫자로 바꿔야하는 것이었다. 이 과정에서 아스키코드로 변환해주는 함수인 ord() 함수를 알게 되었고 이를 사용하여 y좌표를 숫자로 표현할 수 있었다.

 

다른 코드

# 현재 나이트의 위치 입력받기
input_data = input()
row = int(input_data[1])
column = int(ord(input_data[0])) - int(ord('a')) + 1

# 나이트가 이동할 수 있는 8가지 방향 정의
steps = [(-2, -1), (-1, -2), (1, -2), (2, -1), (2, 1), (1, 2), (-1, 2), (-2, 1)]

# 8가지 방향에 대하여 각 위치로 이동이 가능한지 확인
result = 0
for step in steps:
    # 이동하고자 하는 위치 확인
    next_row = row + step[0]
    next_column = column + step[1]
    # 해당 위치로 이동이 가능하다면 카운트 증가
    if next_row >= 1 and next_row <= 8 and next_column >= 1 and next_column <= 8:
        result += 1

print(result)

책의 솔루션에서는 방향벡터를 하나의 리스트에 순서쌍으로 저장하였다. dx, dy가 아닌 다른 방법도 알게 되어 좋았다.

둘 다 자주 쓰이는 방법이므로 둘다 기억해야겠다.

 

---

참고자료: 이것이 취업을 위한 코딩 테스트다 with 파이썬 - 나동빈 지음(한빛미디어)