[백준] 2583: 영역 구하기 python 구현 / BFS

문제

눈금의 간격이 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개의 직사각형 내부를 제외한 나머지 부분이 몇 개의 분리된 영역으로 나누어지는지, 그리고 분리된 각 영역의 넓이가 얼마인지를 구하여 이를 출력하는 프로그램을 작성하시오.

 

풀이

좌표 문제는 항상 인덱싱이 헷갈린다.

이번 문제에서 좌표계방식으로 인덱싱 되어있는 왼쪽 방식에서, python에서 row을 조회할때 사용하는 오른쪽 방식으로 row의 수를 바꿔주어야 한다 !

그리고 좌표가 (4, 0)일 때 모눈 종이의 칸 개수는 4개임을 유의하여 부등호를 작성하여야 한다.

 

풀이의 흐름은 다음과 같다. 

1. 입력 받기
2. vst 만들기 (n+1), (m+1)만큼
3. 사각형 안에 포함되는 칸 방문처리
   
   • input shape: (x1, y1), (x2, y2)
   • input: (0, 2) (4, 4)
   • transform: (0, 3) (4, 1)
   • 이에 따라 y의 부등호도 방향이 바뀌어야 한다.
4. BFS로 탐색

CODE

import sys
from collections import deque

m, n, k = map(int, input().split()) #m:row, n:col, k:사각형 개수
vertex_list = []
for _ in range(k):
    vertex = list(map(int, sys.stdin.readline().split()))
    vertex_list.append(vertex)

# vst 만들기
vst = [[0]*(n+1) for _ in range(m+1)]

def bfs(st_point):
    queue = deque()
    queue.append((st_point[0], st_point[1]))
    vst[st_point[0]][st_point[1]] = 1
    cnt = 1

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

    while queue:
        now = queue.popleft()
        x, y = now  # row, col

        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            # 범위 체크
            if nx<1 or ny<1 or nx>m or ny>n: continue
            # 방문 체크
            if vst[nx][ny]: continue
            # print(nx, ny, m, n)
            queue.append((nx, ny))
            vst[nx][ny] = 1
            cnt += 1

    return cnt

for x in range(1, m+1):  # row
    for y in range(1, n+1):   # col
        for i in vertex_list:
            y1, x1, y2, x2 = i         

            # 사각형 안에 포함되는지 체크
            if y > y1 and y <= y2 and x > (m- x2) and x <= (m - x1):
                vst[x][y] = 1
                continue
        

    
# 방문x, 사각형x
cnt_list = []
for x in range(1, m+1):  # row
    for y in range(1, n+1):   # col
        # 방문 여부 확인
        if vst[x][y]: continue

        st_point = (x, y)
        result_cnt = bfs(st_point)
        cnt_list.append(result_cnt)
            
print(len(cnt_list))
for i in sorted(cnt_list):
    print(i, end=' ')