본문 바로가기
코딩테스트

[이것이 코딩테스트다] Chapter 2 구현 실전문제 왕실의 나이트

by 박매트 2024. 3. 18.

실전문제 - 왕실의 나이트

 

나이트는 말을 타고 있기 때문에 이동을 할 때는 L자 형태로만 이동할 수 있으며 정원 밖으로는 나갈 수 없다.

 

이동하는 법

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

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

 

나이트의 위치가 주어졌을 때 , 나이트가 이동할 수 있는 경우의 수를 출력하는 프로그램

 

입력예시 : a1

출력예씨 : 2

 

나이트의 이동 경로를 steps 변수에 넣는다면 위 규칙에 따라 8가지 경우의 수를 정의할 수 있다.

각 8가지 방향에 대하여 각 위치로 이동이 가능한 지 확인한다.

 

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

 

이전 문제에서는 dx, dy 리스트를 선언하여 이동할 방향을 기록할 수 있도록 한다. 

# L, R, U, D에 따른 이동 방향
dx = [0,0,-1,1]
dy = [-1,1,0,0]
move_types = ['L','R','U','D']

 

 

이번 문제에서는 steps 변수가 dx와 dy 변수의 기능을 대신하여 수행한다.

# 나이트가 이동할 수 있는 8가지 방향 정의
steps = [(-2,-1),(-1,-2),(1,-2),(2,-1),(2,1),(1,2),(-1,2),(-2,1)]