728x90
반응형
문제:
체스판 위에 한 나이트가 놓여져 있다. 나이트가 한 번에 이동할 수 있는 칸은 아래 그림에 나와있다. 나이트가 이동하려고 하는 칸이 주어진다. 나이트는 몇 번 움직이면 이 칸으로 이동할 수 있을까?
입력:
입력의 첫째 줄에는 테스트 케이스의 개수가 주어진다.
각 테스트 케이스는 세 줄로 이루어져 있다. 첫째 줄에는 체스판의 한 변의 길이 l(4 ≤ l ≤ 300)이 주어진다. 체스판의 크기는 l × l이다. 체스판의 각 칸은 두 수의 쌍 {0, ..., l-1} × {0, ..., l-1}로 나타낼 수 있다. 둘째 줄과 셋째 줄에는 나이트가 현재 있는 칸, 나이트가 이동하려고 하는 칸이 주어진다.
출력:
각 테스트 케이스마다 나이트가 최소 몇 번만에 이동할 수 있는지 출력한다.
풀이방법:
너비 우선 탐색의 가장 기초적인 문제다. 이미 지나온 곳을 다시 방문하지 않도록 0으로 초기화된 visited 배열을 만들었고, 현재 stage에서 나이트가 있을 수 있는 위치를 able이라는 배열에 저장하도록 했다.
bfs 함수 내에선 우선 able 내에 목적지가 있는지 확인하는 과정을 먼저 거친다. 있다면 현재 stage(count)를 출력하도록 하고, 없으면 나이트가 움직일 수 있는 경우의 수인 candidate를 able을 각 케이스에 대해 더해줘서 새로운 able을 만들도록 한다. 이 과정을 재귀적으로 반복하면서 값을 찾는다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
candidate = [(2,1),(1,2),(2,-1),(1,-2),(-1,2),(-1,-2),(-2,1),(-2,-1)]
def bfs(able,visited):
global count
if (eX,eY) in able:
print(count)
else:
count+=1
temp = []
for a in able:
for c in candidate:
nx,ny = a[0]+c[0],a[1]+c[1]
if 0<=nx<l and 0<=ny<l and visited[nx][ny]==0:
visited[nx][ny]=1
temp.append((nx,ny))
bfs(temp,visited)
N=int(input())
for _ in range(N):
l = int(input())
sX,sY = map(int,input().split())
eX,eY = map(int,input().split())
visited = [[0 for _ in range(l)]for _ in range(l)]
count=0
able = [(sX,sY)]
visited[sX][sY] = 1
bfs(able,visited)
|
cs |
문제링크:
728x90
반응형
'Algorithm > Python' 카테고리의 다른 글
[BOJ]1747. 소수&팰린드롬 (0) | 2020.11.26 |
---|---|
[BOJ]1707. 이분 그래프 (0) | 2020.11.24 |
[BOJ]1992. 쿼드트리 (0) | 2020.11.17 |
[BOJ]1041. 주사위 (0) | 2020.11.12 |
[BOJ]1059. 수2 (0) | 2020.11.10 |