분류
너비 우선 탐색(bfs), 깊이 우선 탐색(dfs), 그래프 이론(graphs), 그래프 탐색(graph_traversal)
문제 설명
코레스코 콘도미니엄 8층은 학생들이 3끼의 식사를 해결하는 공간이다. 그러나 몇몇 비양심적인 학생들의 만행으로 음식물이 통로 중간 중간에 떨어져 있다. 이러한 음식물들은 근처에 있는 것끼리 뭉치게 돼서 큰 음식물 쓰레기가 된다.
이 문제를 출제한 선생님은 개인적으로 이러한 음식물을 실내화에 묻히는 것을 정말 진정으로 싫어한다. 참고로 우리가 구해야 할 답은 이 문제를 낸 조교를 맞추는 것이 아니다.
통로에 떨어진 음식물을 피해가기란 쉬운 일이 아니다. 따라서 선생님은 떨어진 음식물 중에 제일 큰 음식물만은 피해 가려고 한다.
선생님을 도와 제일 큰 음식물의 크기를 구해서 “10ra"를 외치지 않게 도와주자.
입력
첫째 줄에 통로의 세로 길이 N(1 ≤ N ≤ 100)과 가로 길이 M(1 ≤ M ≤ 100) 그리고 음식물 쓰레기의 개수 K(1 ≤ K ≤ N×M)이 주어진다. 그리고 다음 K개의 줄에 음식물이 떨어진 좌표 (r, c)가 주어진다.
좌표 (r, c)의 r은 위에서부터, c는 왼쪽에서부터가 기준이다. 입력으로 주어지는 좌표는 중복되지 않는다.
3 4 5
3 2
2 2
3 1
2 3
1 1
출력
첫째 줄에 음식물 중 가장 큰 음식물의 크기를 출력하라.
4
내 코드 1(bfs): 틀렸습니다
n, m, k = map(int, input().split())
graph = [[0] * m for _ in range(n)]
visited = [[False] * m for _ in range(n)]
for i in range(k):
x, y = map(int, input().split())
graph[x-1][y-1] = 1
def bfs(x, y):
q = [(x, y)]
cnt = 0
while q:
x, y = q.pop(0)
visited[x][y] = True
for dx, dy in [(1, 0), (-1, 0), (0, 1), (0, -1)]:
nx, ny = x+dx, y+dy
if 0<=nx<n and 0<=ny<m and not visited[nx][ny]:
if graph[nx][ny] == 1:
q.append((nx, ny))
cnt += 1
graph[nx][ny] = cnt
visited[nx][ny] = True
for i in range(n):
for j in range(m):
bfs(i, j)
print(max(map(max, graph)))-
입력으로 들어오는 좌표에 대해서만 1로 초기화
-
(i, j)의 상하좌우에서 방문하지 않았으면서 값이 1인 노드가 있으면 큐에 삽입한 뒤, cnt 값을 증가시키고 해당 노드의 값을 cnt로 변경 및 방문 처리
틀린 부분
-
cnt = 0 → cnt = 1
-
bfs(i, j) 할 때 cnt를 시작할 위치의 값이 1인지 체크하지 않음
반례
그래프가 다음과 같을 때
1 1 1 0 0 0 0 1 0 0 0 1 1 1 1 0 1 0 1 0 1
-
기본 위치, 즉 (i, j)에서 값이 1일 때 기본적으로 음식물이 1개 있는 걸로 시작하므로 cnt = 1로 초기화되어야 함
출력 [1, 2, 4, 0, 0, 0, 0] [1, 0, 0, 0, 1, 3, 4] [3, 0, 1, 0, 2, 0, 5](0, 0)에서 시작했을 때 cnt = 0이면 (1, 0)의 값이 0 + 1로 1이 되면서 개수가 하나씩 부족하게 세어짐
-
cnt는 수정했지만 애초에 그래프 값이 1이어야 탐색 시작하는 거 !! 를 놓쳤음
출력 [1, 3, 5, 0, 0, 0, 0] [2, 0, 0, 0, 2, 4, 5] [4, 0, 2, 0, 3, 0, 6](1, 2)에서 값이 0인데도 bfs 실행되고 한 칸 아래의 값이 1이므로 1+1=2 로 들어가게 됨. (0, 4)에서도 마찬가지
내 코드 2(bfs): 맞았습니다
n, m, k = map(int, input().split())
graph = [[0] * m for _ in range(n)]
visited = [[False] * m for _ in range(n)]
for i in range(k):
x, y = map(int, input().split())
graph[x-1][y-1] = 1
def bfs(x, y):
q = [(x, y)]
cnt = 1
while q:
x, y = q.pop(0)
visited[x][y] = True
for dx, dy in [(1, 0), (-1, 0), (0, 1), (0, -1)]:
nx, ny = x+dx, y+dy
if 0<=nx<n and 0<=ny<m and not visited[nx][ny]:
if graph[nx][ny] == 1:
q.append((nx, ny))
cnt += 1
graph[nx][ny] = cnt
visited[nx][ny] = True
for i in range(n):
for j in range(m):
if graph[i][j] == 1:
bfs(i, j)
print(max(map(max, graph)))다른 사람 코드(bfs)
n, m, k = map(int, input().split())
trash = [[False] * m for _ in range(n)]
for _ in range(k):
row, col = map(int, input().split())
trash[row-1][col-1] = True
size = 0
def bfs(x, y):
global size
q = [(x, y)]
trash[x][y] = False
size = 1
while q:
x, y = q.pop()
for dx, dy in [(1, 0), (-1, 0), (0, 1), (0, -1)]:
nx, ny = nx, ny = x+dx, y+dy
if 0<=nx<n and 0<=ny<m and trash[nx][ny]:
q.append((nx, ny))
trash[nx][ny] = False
size += 1
if __name__ == '__main__':
max_size = 0
for i in range(n):
for j in range(m):
if trash[i][j]:
trash[i][j] = False
size = 0
bfs(i, j)
max_size = max(size, max_size)
print(max_size)-
visited 배열을 따로 선언하지 않고, size 전역 변수를 업데이트 하는 방식
-
방문 처리: 그래프 값 True -> False
dfs 코드(추후 추가)
