이코테 2021, 구현/ 왕실의 나이트(파이썬)

유튜브 참고 : https://www.youtube.com/watch?v=2zjoKjt97vQ

 

문제 정의

행복 왕국의 왕실 정원은 체스판과 같은 8 x 8 좌표 평면이다. 왕실 정원의 특정한 한 칸에서 나이트가 서있다. 나이트는 말을 타고 있기 때문에 이동을 할 때는 L자 형태로만 이동할 수 있으며 정원 밖으로는 나갈 수 없다. 나이트는 특정 위치에서 다음과 같은 2가지 경우로 이동할 수 있다.1. 수평으로 두 칸 이동한 뒤에 수직으로 한 칸 이동하기2. 수직으로 두 칸 이동한 뒤에 수평으로 한 칸 이동하기

 

 

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

 

c2에 있을 때 이동할 수 있는 경우의 수는 6가지이다.

입력

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

 

 

출력

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

 

예제 입력 1

a1

예제 출력 1

2

 

 

접근 방법 

1. 주어진 조건에 맞는 맵(N x N)을 만든다.

2. 이동 조건을 8가지로 분류하고 현재 위치에서 이동할 때 map 내에 몇 개나 가능한 지 체크한 뒤 이를 출력한다.

 

 

코드

night = input()

chessboard = [[ord(col), row] for row in range(1, 9) for col in ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']]


# 위로 두 칸, 좌우로 한 칸씩 / 아래로 두 칸, 좌우로 한 칸씩 
actions = [[2, 1], [2, -1], [-2, 1], [-2, -1], [1, 2], [-1, 2], [1, -2], [-1, -2]]

def count(night):
    col = ord(night[0])
    row = int(night[1])
    result = 0
    for action in actions:
        new_col = action[0] + col
        new_row = action[1] + row
        if [new_col, new_row] in chessboard:
            result += 1
    print(result)

count(night)