목록2667 (1)

코딩 공부

[Python](BFS) 백준 2667번 : 단지번호붙이기

지도 모양을 그대로 arr라는 이차원 배열에 입력한다. 차례로, 방문하지 않았던 곳만을 방문하기 위해서 이차원 배열인 visited 배열도 만들어준다. 각 단지의 집 개수를 모아서 저장하기 위한 size_lst를 list형태로 만들어준다. 이중 for문으로 arr를 탐색하면서 방문한 적이 없으면서 집이 있는 곳(1로 표시된 곳)이 발견되면 bfs를 돌린다. (이때 새로운 단지가 발견되는 것이므로 town에 +1을 해주고, bfs의 결과값으로 집의 개수를 return해 size_lst에 추가해준다. from collections import deque def bfs(si, sj): q = deque() q.append((si, sj)) visited[si][sj] = 1 count = 1..
BFS & DFS 2024. 1. 2. 01:35
이전 Prev 1 Next 다음