문제
어떤 큰 도화지에 그림이 그려져 있을 때, 그 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하여라. 단, 그림이라는 것은 1로 연결된 것을 한 그림이라고 정의하자. 가로나 세로로 연결된 것은 연결이 된 것이고 대각선으로 연결이 된 것은 떨어진 그림이다. 그림의 넓이란 그림에 포함된 1의 개수이다.
입력
첫째 줄에 도화지의 세로 크기 n(1 ≤ n ≤ 500)과 가로 크기 m(1 ≤ m ≤ 500)이 차례로 주어진다. 두 번째 줄부터 n+1 줄 까지 그림의 정보가 주어진다. (단 그림의 정보는 0과 1이 공백을 두고 주어지며, 0은 색칠이 안된 부분, 1은 색칠이 된 부분을 의미한다)
출력
첫째 줄에는 그림의 개수, 둘째 줄에는 그 중 가장 넓은 그림의 넓이를 출력하여라. 단, 그림이 하나도 없는 경우에는 가장 넓은 그림의 넓이는 0이다.
DFS
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import sys
sys.setrecursionlimit(10000000)
input = sys.stdin.readline
n,m = map(int,input().split())
graph = []
for i in range(n):
graph.append(list(map(int,input().split())))
# 이 방향벡터 사용해서 for문 돌리면 메모리 초과
#dx = [-1,1,0,0]
#dy = [0,0,-1,1]
areas = []
cnt=0
def dfs(x,y):
global cnt
if x <= -1 or x >= n or y <= -1 or y >= m:
return False
if graph[x][y] == 1:
cnt+=1
graph[x][y] = 0
dfs(x + 1, y)
dfs(x - 1, y)
dfs(x, y + 1)
dfs(x, y - 1)
# 방향벡터로 for문 돌리면 메모리 초과
# for i in range(4):
# nx = x+dx[i]
# ny = y+dy[i]
# dfs(nx,ny)
return True
return False
for i in range(n):
for j in range(m):
#1일때만
if graph[i][j]==1:
cnt=0
dfs(i,j)
areas.append(cnt)
if len(areas)==0:
print(0,0,sep='\n')
else:
print(len(areas))
print(max(areas))
BFS
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
from collections import deque
# n,m = 6,5
# graph = [[1,1,0,1,1],
# [0,1,1,0,0],
# [0,0,0,0,0],
# [1,0,1,1,1],
# [0,0,1,1,1],
# [0,0,1,1,1]]
n,m = map(int,input().split())
graph = [list(map(int,input().split())) for _ in range(n)]
#그림을 세야하고
#1의 개수가 가장 많은 것을 출력해야함
dx = [-1,1,0,0]
dy = [0,0,-1,1]
def bfs(x,y,cnt):
graph[x][y]=0
queue = deque()
queue.append([x,y])
cnt+=1
while queue:
a,b=queue.popleft()
for i in range(4):
nx = a+dx[i]
ny = b+dy[i]
if -1<nx<n and -1<ny<m and graph[nx][ny] == 1:
graph[nx][ny] = 0
queue.append([nx,ny])
cnt+=1
return cnt
result = []
for i in range(n):
for j in range(m):
if graph[i][j] == 1:
result.append(bfs(i,j,0))
if len(result)==0:
print(0,0,sep='\n')
else:
print(len(result))
print(max(result))
처음에 DFS로 풀었는데 제출하니 시간 초과가 나서 BFS로 풀었다.
근데 풀고 나서도 DFS로 안풀린게 찝찝해서 계속 붙잡고 해보니, 방향벡터를 사용 안하면 메모리 초과가 안 난다.
DFS
BFS
메모리 사용량을 보면 DFS가 압도적으로 크다.
확실히 경우에 따라서 모든 경우의 수를 살펴봐야 할 수도 있기 때문에 DFS는
DFS는 재귀방식으로 수행되고, 이에 따라 함수를 호출할때마다 스택이 쌓이기 때문에 메모리를 많이 차지하게 된다.
반면 BFS는 queue에 쌓이는 컬랙션이나 class 객체의 크기가 크지 않다면 훨씬 적은 비용이 발생할 수 있다.
따라서 BFS가 메모리 측면으로는 유리할 수 있다.