백준 2583: 영역 구하기 - BFS, DFS(파이썬)

영역 구하기 성공

2583번: 영역 구하기

문제

눈금의 간격이 1인 M×N(M,N≤100)크기의 모눈종이가 있다. 이 모눈종이 위에 눈금에 맞추어 K개의 직사각형을 그릴 때, 이들 K개의 직사각형의 내부를 제외한 나머지 부분이 몇 개의 분리된 영역으로 나누어진다.

예를 들어 M=5, N=7 인 모눈종이 위에 <그림 1>과 같이 직사각형 3개를 그렸다면, 그 나머지 영역은 <그림 2>와 같이 3개의 분리된 영역으로 나누어지게 된다.

<그림 2>와 같이 분리된 세 영역의 넓이는 각각 1, 7, 13이 된다.

M, N과 K 그리고 K개의 직사각형의 좌표가 주어질 때, K개의 직사각형 내부를 제외한 나머지 부분이 몇 개의 분리된 영역으로 나누어지는지, 그리고 분리된 각 영역의 넓이가 얼마인지를 구하여 이를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 M과 N, 그리고 K가 빈칸을 사이에 두고 차례로 주어진다. M, N, K는 모두 100 이하의 자연수이다. 둘째 줄부터 K개의 줄에는 한 줄에 하나씩 직사각형의 왼쪽 아래 꼭짓점의 x, y좌표값과 오른쪽 위 꼭짓점의 x, y좌표값이 빈칸을 사이에 두고 차례로 주어진다. 모눈종이의 왼쪽 아래 꼭짓점의 좌표는 (0,0)이고, 오른쪽 위 꼭짓점의 좌표는(N,M)이다. 입력되는 K개의 직사각형들이 모눈종이 전체를 채우는 경우는 없다.

출력

첫째 줄에 분리되어 나누어지는 영역의 개수를 출력한다. 둘째 줄에는 각 영역의 넓이를 오름차순으로 정렬하여 빈칸을 사이에 두고 출력한다.

 

내 코드

import sys
sys.setrecursionlimit(10**6)

def dfs(x, y):
    #해당 좌표 부분 방문 처리 및 넓이++;
    global area
    area+=1 
    myMap[x][y]=1

    for i in range(4):
        nx=x+dx[i]
        ny=y+dy[i]

        if nx<0 or nx>=m or ny<0 or ny>=n:
            continue
        if myMap[nx][ny]==0:
            dfs(nx,ny)



#메인 코드 부분
#북동남서
dx=[-1,0,1,0]
dy=[0,1,0,-1]

#m:세로, n:가로
m,n,k=map(int,sys.stdin.readline().split())

#전체 크기의 배열을 만들고 직사각형 위치 시키기
myMap=[[0]*n for _ in range(m)]
for _ in range(k):
    y1,x1,y2,x2=map(int,sys.stdin.readline().split())

    for i in range(x2-x1):
        for j in range(y2-y1):
            myMap[x1+i][y1+j]=1


areaList=[]
cnt=0
for i in range(m):
    for j in range(n):
        if myMap[i][j]==0:
            area=0 #특정 좌표에서 BFS 시작전에 넓이 0 초기화.
            dfs(i,j)
            cnt+=1 #DFS할 때마다 영역 개수++;
            areaList.append(area) #DFS가 끝나면 해당 영역 넓이 저장.

print(cnt)
print(*sorted(areaList))

 

풀이 및 접근)

- 일반적인 BFS, DFS의 성격을 띠는 문제이다. 인접한 영역들을 한 덩어리로 묶어서 그 덩어리들의 개수와 덩이리별 개수를 세는 문제이다.

- 약간 신경써야 할 것은 좌표를 어떻게 처리할 것이냐이다. 배열이 arr[x][y] 꼴 일때, x 부분이 세로 부분이고, y 부분이 가로 부분이다. 하지만 문제에서는 x가 가로 부분이고, y가 세로 부분이다. 이를 고려하여 배열 부분을 거꾸로 할지, 문제에서의 부분을 거꾸로할지 정하면 된다. 정한 방법의 것으로 해 나가면 된다.

- 내 풀이의 경우, 문제에서 주어진 (x,y)를 배열에 적용시킬 때 (y1, x1) (y2, x2)로 받아서 arr[x][y]꼴로 저장되게 하였다.

- 직사각형이 있는 곳을 1로 만들고 0인 부분에서 DFS를 시작하도록 하였다. 방문을 했으면 1로 그 값을 바꿔주면 돼서 방문여부를 저장할 visited[][]와 같은 배열은 필요없다.

- 나머지 부분은 주석을 보면 이해가 될 것이다.