[이코테] 왕실의 나이트

출처: 이것이 취업을 위한 코딩테스트다 with 파이썬 (나동빈)

https://www.youtube.com/watch?v=2zjoKjt97vQ&list=PLRx0vPvlEmdAghTr5mXQxGpHjWqSz0dgC&index=2

 

문제

• 행복 왕국의 왕실 정원은 체스판과 같은 8*8좌표 평면입니다.  왕실 정원의 특정한 한 칸에 나이트가 서 있습니다. 나이트는 매우 충성스러운 신하로서 매일 무술을 연마합니다.
• 나이트는 말을 타고 있기 때문에 이동을 할 때는 L자 형태로만 이동할 수 있으며 정원 밖으로는 나갈 수 없습니다.
• 나이트는 특정 위치에서 다음과 같은 2가지 경우로 이동할 수 있습니다.
  1. 수평으로 두 칸 이동한 뒤에 수직으로 한 칸 이동하기
  2. 수직으로 두 칸 이동한 뒤에 수평으로 한 칸 이동하기
• 이처럼 8*8 좌표 평면상에서 나이트의 위치가 주어졌을 때 나이트가 이동할 수 있는 경우의 수를 출력하는 프로그램을 작성하세요. 왕실의 정원에서 행 위치를 표현할 때는 1부터 8로 표현하며, 열 위치를 표현할 때는 a부터 h로 표현합니다.

 

문제 조건

• 풀이 시간: 20분
• 시간제한: 1초
• 메모리 제한: 128MB
• 입력 조건: 첫째 줄에 8*8 좌표 평면상에서 현재 나이트가 위치한 곳의 좌표를 나타내는 두 문자로 구성된 문자열이 입력된다. 입력 문자는 a1처럼 열과 행으로 이루어진다.
• 출력 조건: 첫째 줄에 나이트가 이동할 수 있는 경우의 수를 출력하시오.

---

작성 코드

col=['a','b','c','d','e','f','g','h']
dx=[-1,1,-2,-2,-1,1,2,2]
dy=[-2,-2,-1,1,2,2,-1,1]

str = input()

#current position: x,y

for i in range(len(col)):
  if str[0]==col[i]:
    x = i
    
y = int(str[1])

#start counting
cnt=0
for i in range(len(dx)):
  nx = x+dx[i]
  ny = y+dy[i]
  if 1<=nx<=8 and 1<=ny<=8:
    cnt += 1

#print result
print(cnt)

 

강의 예시 코드

강의에서는 방향벡터를 정의할 때 dx, dy로 나누어 정의하지 않고 이차원 배열에 한 번에 저장했다.

또 문자 a~h를 아스키코드상 숫자로 변환, 'a'를 빼주고 1을 더해서 해당 문자가 몇 번째 column인지 바로 구했다.

 

 

방향벡터를 2차원 배열에 저장하니까 가독성이 훨씬 좋은 것 같다