본문 바로가기

알고리즘 (Python)/이것이 코딩 테스트다 with 파이썬

[구현 알고리즘] 왕실의 나이트 - 파이썬(python)

728x90
반응형

왕실의 나이트

난이도 : 下 풀이 시간 : 20분

시간 제한 : 1초 메모리 제한 : 128 MB

 


 

해답

 

location = input()
count = 0

x = ord(location[0]) - 96
y = int(location[1])

steps = [
    (-2, -1), (-2, 1), (-1, -2), (-1, 2),
    (1, -2), (1, 2), (2, -1), (2, 1)
]

for step in steps:
    nx = x + step[0]
    ny = y + step[1]

    if 1<=nx<=8 and 1<=ny<=8:
        count = count + 1

print(count)

예시

 

# 현재 나이트의 위치 입력받기
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 리스트를 선언하여 이동할 방향을 기록할 수 있도록 하였음
이번 소스코드에서는 steps 변수가 dx와 dy 변수이 기능을 대신하여 수행함
728x90
반응형