728x90
헌내기는 친구가 필요해 성공
시간 제한메모리 제한제출정답맞힌 사람정답 비율
1 초 (하단 참고) | 1024 MB | 3965 | 2297 | 1923 | 60.605% |
문제
2020년에 입학한 헌내기 도연이가 있다. 도연이는 비대면 수업 때문에 학교에 가지 못해 학교에 아는 친구가 없었다. 드디어 대면 수업을 하게 된 도연이는 어서 캠퍼스 내의 사람들과 친해지고 싶다.
도연이가 다니는 대학의 캠퍼스는 크기이며 캠퍼스에서 이동하는 방법은 벽이 아닌 상하좌우로 이동하는 것이다. 예를 들어, 도연이가 (x,y) 에 있다면 이동할 수 있는 곳은 (x+1,y), (x-1,y), (x,y+1), (x,y-1) 이다. 단, 캠퍼스의 밖으로 이동할 수는 없다.
불쌍한 도연이를 위하여 캠퍼스에서 도연이가 만날 수 있는 사람의 수를 출력하는 프로그램을 작성해보자.
입력
첫째 줄에는 캠퍼스의 크기를 나타내는 두 정수 N,M (0 < N,M <= 600)이 주어진다.
둘째 줄부터 N개의 줄에는 캠퍼스의 정보들이 주어진다. O는 빈 공간, X는 벽, I는 도연이, P는 사람이다. I가 한 번만 주어짐이 보장된다.
출력
첫째 줄에 도연이가 만날 수 있는 사람의 수를 출력한다. 단, 아무도 만나지 못한 경우 TT를 출력한다.
예제 입력 1 복사
3 5
OOOPO
OIOOX
OOOXP
예제 출력 1 복사
1
예제 입력 2 복사
3 3
IOX
OXP
XPP
예제 출력 2 복사
TT
처음 이 문제를 풀땐, DFS알고리즘이 가장 적합하다고 생각했지만, 실제로 이 문제를 풀땐 BFS알고리즘으로 푸는게 가장 편하다.
BFS알고리즘은 넓이우선탐색으로, 길찾기에 유용한것같다.
또한, BFS알고리즘에 많이 약해서 이번기회에 공부를 더 해보려고 한다.
#21736 헌내기는 친구가 필요해
import sys
from collections import deque
n,m = map(int,sys.stdin.readline().split())
li = []
for i in range(n):
li.append(list(sys.stdin.readline().rstrip()))
dx = [-1,1,0,0]
dy = [0,0,-1,1]
value = []
def bfs(x,y):
queue = deque()
queue.append([x,y])
while queue:
x,y = queue.popleft()
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if nx < 0 or nx >= n or ny < 0 or ny >= m:
continue
if li[nx][ny] == 'X':
continue
if li[nx][ny] == 'O' or li[nx][ny] == 'P':
if li[nx][ny] == 'P':
value.append([nx,ny])
li[nx][ny] = 'V'
queue.append([nx,ny])
for i in range(n):
for j in range(m):
if li[i][j] == 'I':
bfs(i,j)
break
if len(value) > 0:
print(len(value))
else:
print('TT')
728x90
'Study > CodingTest' 카테고리의 다른 글
[백준/Python] (S2)1260 - DFS와 BFS (0) | 2023.12.07 |
---|---|
[백준/Python] (S1)14940 - 쉬운 최단거리 (0) | 2023.08.25 |
[백준/Python] (G5)10026 - 적록색약 (0) | 2023.07.26 |
[백준/Python] (G5)5430 - AC (0) | 2023.07.25 |
[백준/Python] (S4)1755 - 숫자놀이 (0) | 2023.07.14 |