<청춘> 격정적으로 사는 것

밤을 새고 공부한 다음 날 새벽에 느꼈던 생생한 환희와 야생적인 즐거움을 잊을 수 없다

코딩테스트

[구현] 실전 문제 <2> 왕실의 나이트 / 이것이 취업을 위한 코딩테스트다 with 파이썬

수학도 2021. 8. 3. 17:05

왕실의 나이트

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

 

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

  • 수평으로 두 칸 이동한 뒤에 수직으로 한 칸 이동하기
  • 수직으로 두 칸 이동한 뒤에 수평으로 한 칸 이동하기

 

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

 

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

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

또 다른 예로 나이트가 c2에 위치해 있다면 나이트가 이동할 수 있는 경우의 수는 6가지이다.

 

입력 조건

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

 

출력 조건

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

 

입력 예시

a1

 

출력 예시

2

소스코드

xy = input()
x = int(xy[1])
y = int(ord(xy[0])) - int(ord('a')) + 1

directions = [(-2, -1), (-2, 1), (2, -1), (2, 1), (1, -2), (1, 2), (-1, -2), (-1, 2)]
count = 0

# 이동할 좌표 확인
for d in directions:
    nowx = x + d[0]
    nowy = y + d[1]
    # 정원을 벗어나면 이동하지 않는다.
    if nowx < 1 or nowx > 8 or nowy < 1 or nowy > 8:
        continue
    # 벗어나지 않으면 경우의 수 증가
    count += 1

print(count)

 

해설

ord(문자) : 문자의 아스키코드 값을 반환한다.

ord('a') = 97

ord('d') = 100

 

a1을 입력하면 xy = a1 이므로 xy[0] = a 이고 xy[1] = 1이 된다.

x = int(xy[1]) = int(1) = 1

y = int(ord(xy[0])) - int(ord('a')) + 1 = int(ord('a')) - int(ord('a')) + 1 = 97 - 97 + 1 = 1