알고리즘/문제풀이

DFS(그림/S1) - 1926

문승주 2023. 2. 24. 13:16
반응형

1. 문제

어떤 큰 도화지에 그림이 그려져 있을 때, 그 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하여라. 단, 그림이라는 것은 1로 연결된 것을 한 그림이라고 정의하자. 가로나 세로로 연결된 것은 연결이 된 것이고 대각선으로 연결이 된 것은 떨어진 그림이다. 그림의 넓이란 그림에 포함된 1의 개수이다.

 

2. 입력

첫째 줄에 도화지의 세로 크기 n(1 ≤ n ≤ 500)과 가로 크기 m(1 ≤ m ≤ 500)이 차례로 주어진다. 두 번째 줄부터 n+1 줄 까지 그림의 정보가 주어진다. (단 그림의 정보는 0과 1이 공백을 두고 주어지며, 0은 색칠이 안된 부분, 1은 색칠이 된 부분을 의미한다)

 

6 5
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

 

3. 출력

첫째 줄에는 그림의 개수, 둘째 줄에는 그 중 가장 넓은 그림의 넓이를 출력하여라. 단, 그림이 하나도 없는 경우에는 가장 넓은 그림의 넓이는 0이다.

 

4
9

 

4. 코드

import sys
sys.setrecursionlimit(10**6)

input = sys.stdin.readline
n,m = map(int, input().split())
paint = [list(map(int,input().split())) for _ in range(n)]
mx = [0,0,-1,1]
my = [-1,1,0,0]
area = 0
areaList = []

def dfs(x,y):
  paint[x][y] = 0
  global area
  area += 1
  # 상화좌우 탐색
  for i in range(4):
    dx,dy = x+mx[i],y+my[i]
    if dx < 0 or dy < 0 or dx >= n or dy >= m:
      continue
    if paint[dx][dy] == 1:
      dfs(dx,dy)
    
for i in range(n):
  for j in range(m):
    # 1인 경우 함수 호출
    if paint[i][j] == 1:
      area = 0
      dfs(i,j)
      areaList.append(area)

# 결과 출력
print(len(areaList))
if len(areaList) > 0:
  print(max(areaList))
else:
  print(0)

 

5. 풀이설명

반응형