[이코테] 구현 문제 : 왕실의 나이트

문제

행복 왕국의 왕실 정원은 체스판과 같은 8 x 8 죄표 평면이다. 왕실 정원의 특정한 한 칸에 나이트가 서 있다. 나이트는 매우 충성스러운 신하로서 매일 무술을 연마한다.
나이트는 말을 타고 있기 때문에 이동을 할 매는 L자 형태로만 이동할 수 있으며 정원 밖으로는 나갈 수 없다. 나이트는 특정한 위치에서 다음과 같은 2가지 경우로 이동할 수 있다.

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

이처럼 8 x 8 좌표 평면상에서 나이트의 위치가 주어졌을 때 나이트가 이동할 수 있는 경우의 수를 출력하는 프로그램을 작성하시오. 이때 왕실의 정원에서 행 위치를 표현할 때는 1부터 8로 표현하며, 열 위치를 표현할 때는 a부터 h로 표현한다.
예를 들어 만약 나이트가 a1에 있을 때 이동할 수 있는 경우의 수는 다음 2가지이다. a1의 위치는 좌표 평면에서 구석의 위치에 해당하며 나이트는 정원의 밖으로는 나같 수 없기 때문이다.

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

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

 

입력 조건

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

출력 조건

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

 

입력 예시

a1

출력 예시

2

 

---

<내가 푼 방법>

일단 입력 문자를 받아와서 좌표 상에서 어디에 위치하는지를 숫자로 표시한다. a->1, b->2, ..., h->8 식으로 바꾼다.

그런 다음 이동할 수 있는 8가지 방향을 모두 trans에 저장하고, 그 숫자가 유효한지(1~8사이인지) 확인한다. 유효한 경우만 count에 센다.

str = input()
row = {'a':1, 'b':2, 'c':3, 'd':4, 'e':5, 'f':6, 'g':7, 'h':8}

x = int(str[1])
y = row[str[0]]

trans = [[x+2,y+1],[x+2,y-1],[x-2,y+1],[x-2,y-1],[x+1,y+2],[x+1,y-2],[x-1,y+2],[x-1,y-2]]

count = 0
for i in trans:
  if (1<=i[0]<=8) and (1<=i[1]<=8):
    count += 1

print(count)

풀긴 했는데 코드가 길고 비효율적인 것 같다. 책에는 어떻게 나와있는지 봐보자.

 

<책>

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

저번에 푼 상하좌우 문제와 비슷한 형식으로 풀었다. 이동할 방향을 변수로 정의해두고, 반복문에서 다음에 이동할 위치를 설정하고, 그 위치로 이동이 가능하면 이동하도록 했다. 이런 방식으로 코드를 많이 짜는 것 같으니, 잘 외워둬야 겠다!