교재 : 이것이 코딩 테스트다 with 파이썬
CHAPTER 4 구현
실전문제 4-3 왕실의 나이트 114p
행복 왕국의 왕실 정원은 체스판과 같은 8 × 8 좌표 평면이다. 왕실 정원의 특정한 한 칸에 나이트가 서있다.
나이트는 매우 충성스러운 신하로서 매일 무술을 연마한다
나이트는 말을 타고 있기 때문에 이동을 할 때는 L자 형태로만 이동할 수 있으며 정원 밖으로는 나갈 수 없다
나이트는 특정 위치에서 다음과 같은 2가지 경우로 이동할 수 있다
이처럼 8 × 8 좌표 평면상에서 나이트의 위치가 주어졌을 때 나이트가 이동할 수 있는 경우의 수를 출력하는
프로그램을 작성하라. 왕실의 정원에서 행 위치를 표현할 때는 1부터 8로 표현하며, 열 위치를 표현할 때는
a 부터 h로 표현한다
첫째 줄에 8x8 좌표 평면상에서 현재 나이트가 위치한 곳의 좌표를 나타내는 두 문자로 구성된 문자열이 입력된다. 입력 문자는 a1 처럼 열과 행으로 이뤄진다.
첫째 줄에 나이트가 이동할 수 있는 경우의 수를 출력하시오.
입력 예시
a1
출력 예시
2
dot = list(input())
x = ord(dot[0])-96
y = int(dot[1])
cnt = 0
dx = [-1,1,2,2,1,-1,-2,-2]
dy = [2,2,1,-1,-2,-2,-1,1]
for i in range(len(dx)):
nx = x+dx[i]
ny = y+dy[i]
if nx<=8 and nx>=1 and ny>=1 and ny<=8:
cnt+=1
print(cnt)
책에 있는 풀이는 방향 정의를 한 배열에 Set으로 설정해주었다. 그리고 그 스탭별로 for문에서 불러와서 계산해줬다. 같은 방법이지만 List와 Set을 저렇게 중첩해서 사용할 수 있다는 걸 알게 되어서 또 새로운 풀이 방식을 하나 알아가는 것 같아서 좋았다. 그리고 상하좌우 문제에서 개선하고 싶었던 부분을 바로 개선해서 문제에서 대입해서 풀었는데 풀이 방식이 잘 맞아 떨어져서 너무 뿌듯하다.
# 현재 나이트의 위치 입력받기
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)
- 풀이시간 : 20분
이것이 코딩 테스트다 with 파이썬 - 나동빈 저