[백준] [파이썬] [DFS] 1926번: 그림

문제 보러 가기

문제

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

입력

첫째 줄에 도화지의 세로 크기 n(1 ≤ n ≤ 500)과 가로 크기 m(1 ≤ m ≤ 500)이 차례로 주어진다. 두 번째 줄부터 n+1 줄 까지 그림의 정보가 주어진다. (단 그림의 정보는 0과 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

예제 출력 1

4
9

---

너무 전형적인 DFS문제 기억하는데로 패턴화된 DFS 방식대로 풀었으나 자꾸 메모리초과, 타임에러등이 나왔다.
처음에는 다음과 같이 풀었다.

---

import sys 

dx=[0,-1,0,1]
dy=[-1,0,1,0]

def DFS(x,y):
    global cnt 
    cnt += 1 
    board[x][y]=0
    for i in range(4):
        nx=x+dx[i]
        ny=y+dy[i]
        if 0<=nx<N and 0<=ny<M and board[nx][ny]==1:
            DFS(nx,ny)

if __name__=="__main__":
    N,M = map(int, input().split())
    board=[list(map(int, input().split())) for _ in range(N)]
    count=0
    res=0
    for i in range(N):
        for j in range(M):
            if board[i][j]==1:
                count+=1
                cnt=0
                DFS(i,j)
                if cnt>res:
                    res=cnt 
    print(count)
    print(res)

---

테스트셋에는 맞는데 계속 틀렸다고 나왔다.
알고보니 DFS문제는 sys.setrecursionlimit(10000000) 를 추가해야 한다더라.

저걸 외우기 위해서 sys. Set recursion limit로 외우자

import sys 
sys.setrecursionlimit(10000000)

dx=[0,-1,0,1]
dy=[-1,0,1,0]

def DFS(x,y):
    global cnt 
    cnt += 1 
    board[x][y]=0
    for i in range(4):
        nx=x+dx[i]
        ny=y+dy[i]
        if 0<=nx<N and 0<=ny<M and board[nx][ny]==1:
            DFS(nx,ny)

if __name__=="__main__":
    N,M = map(int, input().split())
    board=[list(map(int, input().split())) for _ in range(N)]
    count=0
    res=0
    for i in range(N):
        for j in range(M):
            if board[i][j]==1:
                count+=1
                cnt=0
                DFS(i,j)
                if cnt>res:
                    res=cnt 
    print(count)
    print(res)

기억하지 set recursion limit

---

• 23.08.29 수정

만약 이런 문제를 DFS로 풀지 못하였다면 BFS로 풀어보자

import sys 
from collections import deque
sys.stdin=open('input.txt', 'r')

dx=[0,-1,0,1]
dy=[-1,0,1,0]

if __name__=="__main__":
    N,M = map(int, input().split())
    board=[list(map(int, input().split())) for _ in range(N)]
    count=0
    res=0

    for i in range(N):
        for j in range(M):
            if board[i][j]==1:
                count+=1
                cnt=1
                Q=deque()
                Q.append((i,j))
                board[i][j]=0
                while Q:
                    x,y=Q.popleft()
                    for k in range(4):
                        nx=x+dx[k]
                        ny=y+dy[k]
                        if 0<=nx<N and 0<=ny<M and board[nx][ny]==1:
                            Q.append((nx,ny))
                            board[nx][ny]=0
                            cnt+=1

                if cnt>res:
                    res=cnt 
    print(count)
    print(res)