[이코테] 왕실의 나이트

문제

행복 왕국의 왕실 정원은 체스판과 같은 8 x 8 좌표 평면이다. 왕실 정원의 특정한 한 칸에 나이트가 서 있다. 나이트는 매우 충성스러운 신하로서 매일 무술을 연마한다.

나이트는 말을 타고 있기 때문에 이동을 할 때는 L자 형태로만 이동할 수 있으며 정원 밖으로는 나갈 수 없다. 나이트는 특정한 위치에서 다음과 같은 2가지 경우로 이동할 수 있다.

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

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

예를 들어 만약 나이트가 a1에 있을 때 이동할 수 있는 경우의 수는 다음 2가지이다. a1의 위치는 좌표 평면에서 구석의 위치에 해당하며 나이트는 정원의 밖으로는 나갈 수 없기 때문이다.

1. 오른쪽으로 두 칸 이동 후 아래로 한 칸 이동하기 (c2)
2. 아래로 두 칸 이동 후 오른쪽으로 한 칸 이동하기 (b3)

또 다른 예로 나이트가 c2에 위치해 있다면 나이트가 이동할 수 있는 경우의 수는 6가지이다. 이건 직접 계산해보시오.

입력

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

출력

첫째 줄에 나이트가 이동할 수 있는 경우의 수를 출력하시오.

예제

Example 1:

Input: 
a1
Output: 
2

조건

시간 제한 : 1초

메모리 제한 : 128 MB

풀이과정

내 풀이1

state = str(input().rstrip())

n = str(state[0])
m = str(state[1])
n = ord(n)
m = int(m)
count = 0

#위로 
if 1 <= m-2 and m-2 <= 8:
  if 97 <= n-1 and n-1 <= 104:
    count += 1
  if 97 <= n+1 and n+1 <= 104:
    count += 1
#아래로
if 1 <= m+2 and m+2 <= 8:
  if 97 <= n-1 and n-1 <= 104:
    count += 1
  if 97 <= n+1 and n+1 <= 104:
    count += 1 
#좌로
if 97<= n-2 and n-2 <=104:
  if 1 <= m-1 and m-1 <= 8:
    count+=1
  if 1 <= m+1 and m+1 <= 8:
    count+=1
#우로
if 97<= n+2 and n+2 <=104:
  if 1 <= m-1 and m-1 <= 8:
    count+=1
  if 1 <= m+1 and m+1 <= 8:
    count+=1
print(count)

모든 경우의 수에 대해서 나누어 계산하였다.

내 풀이 2

s = (input().rstrip())

count = 0

column = s[1]
row = ord(s[0]) - ord('a') + 1

steps = [(-2,-1), (-2, 1),(2, -1), (2, 1), (1, 2), (1, -2), (-1, 2), (-1, 2) ]

count = 0
for step in steps:
  new_column = int(column) + step[0]
  new_row = int(row) + step[1]
  if new_column < 1 or new_column > 8 or new_row < 1 or new_row > 8:
    continue
  count += 1
print(count)

책을 보고 공부한 뒤, 조금 더 깔끔하게 수정한 코드.

책 풀이

나이트가 이동할 수 있는 경로를 하나씩 확인하여 이동하면 된다. 다만, 8 x 8 좌표 평면을 벗어나지 않도록 꼼꼼하게 검사하는 과정이 필요하다.

나이트는 2가지 경로로 움직일 수 있다고 했다.

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

나이트의 이동 경로를 steps 변수에 넣는다면, 이 2가지 규칙에 따라 steps = [(-2, -1), (-1,-2), (1,-2), (2, -1), (2, 1), (1, 2), (-1, 2), (-2, 1)]로 값을 대입할 수 있다.

이제 나이트의 현재 위치가 주어지면 현재 위치에서 이동 경로를 더한 다음, 8 x 8 좌표 평면에 있는지 확인하면 된다. 이 과정은 반복문으로 처리할 수 있다.

조금 더 까다롭게 문제를 출제한다면 입력 문자가 열과 행이 아닌 1a와 같은 행과 열 형태로 들어왔을 때의 예외 처리를 요구할 수도 있다. 이런 다양한 구현 유형에 대비하기 위해서 파이썬 문법을 자유롭게 사용할 수 있도록 훈련하는 것이 중요하다.

다음 답안 예시에서는 나이트가 이동할 수 있는 경로를 steps 변수에 하나씩 담은 것을 확인할 수 있다. 이러한 8가지 경우의 수를 반복문을 이용하여 하나씩 검사한다.

# 현재 나이트의 위치 입력받기
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)