[DFS/BFS]99클럽 코테 스터디 33일차 TIL + 백준/Silver/2583. 영역 구하기

[Silver I] 영역 구하기 - 2583

문제 링크

성능 요약

메모리: 31120 KB, 시간: 64 ms

분류

너비 우선 탐색, 깊이 우선 탐색, 그래프 이론, 그래프 탐색

제출 일자

2024년 8월 25일 22:13:17

문제 설명

눈금의 간격이 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개의 직사각형들이 모눈종이 전체를 채우는 경우는 없다.

출력

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

 

 

<내 코드>

def adj_list(M, N, square_areas):
    adj = [[0 for col in range(N)] for row in range(M)]
    while square_areas :
        square_area = square_areas.pop()
        x1 = square_area[0]
        y1 = square_area[1]
        x2 = square_area[2]
        y2 = square_area[3]

        for row in range(M):
            for col in range(N):
                # 조건에서 착각함
                if y1 <= row < y2 and x1 <= col < x2:
                    adj[row][col] = 1
    return adj

def find_area(grid, M, N):
    area_count = 0
    directions = [(-1, 0), (1, 0), (0, -1), (0, 1)]

    def dfs(grid, row, col):
        stack = [(row, col)]
        area_size = 0

        while stack:
            cy, cx = stack.pop()
            if grid[cy][cx] == 1:
                continue

            grid[cy][cx] = 1
            area_size += 1
            for dy, dx in directions:
                ny, nx = cy + dy, cx + dx
                if (0 <= ny < M) and (0 <= nx < N) and (grid[ny][nx] == 0):
                    stack.append((ny,nx))
        return area_size

    # print(grid)
    area_sizes = []
    for row in range(M):
        for col in range(N):
            if grid[row][col] == 0:
                area_count += 1
                # print(grid)
                area_size = dfs(grid, row, col)
                area_sizes.append(area_size)
    return area_count, area_sizes

if __name__ == "__main__":

    M, N, K = map(int, input().split())
    square_area = []
    for _ in range(K):
        square_area.append(list(map(int, input().split())))

    adj = adj_list(M,N,square_area)
    result, sizes = find_area(adj,M,N)
    sizes.sort()
    sizes = list(map(str, sizes))
    text = ' '.join(sizes)

    print(result)
    print(text)

- grid 관련 DFS문제를 여러번 푸니까 그 감각이 점점 살아났음

 

- grid 같은 문제는 늘 x,y 축이 헷갈렸는데 가장 간단한 방법은 반대로 yx yx yx yx 로 기억하기임

또한 row는 y, col은 x 로 생각하면 편함 결국 yx yx yx임

 

- 혼자서 푸니까 매우 짜릿했음

특히 grid 관련 dfs에서 cx,cy -> dx,cy -> nx,ny 로 넘어가는 것에 느낌이 오게 됨

 

- 오타가 있어서 예상치 못하게 틀렸음 조심하자

 

- 문자열 정리 .join() 을 잊었음 잘 챙기자

 

- map으로 input() 챙길 때 에는 list()를 꼭!!

 

<모범사례>

def dfs(grid, start, M, N):
    stack = [start]
    directions = [(-1, 0), (1, 0), (0, -1), (0, 1)]
    area_size = 0
    
    while stack:
        y, x = stack.pop()
        if grid[y][x] == 1:
            continue
        
        grid[y][x] = 1
        area_size += 1
        
        for dy, dx in directions:
            ny, nx = y + dy, x + dx
            if 0 <= ny < M and 0 <= nx < N and grid[ny][nx] == 0:
                stack.append((ny, nx))
    
    return area_size

def calculate_areas(M, N, rectangles):
    grid = [[0] * N for _ in range(M)]
    
    for x1, y1, x2, y2 in rectangles:
        for y in range(y1, y2):
            for x in range(x1, x2):
                grid[y][x] = 1
    
    area_sizes = []
    for y in range(M):
        for x in range(N):
            if grid[y][x] == 0:
                area_size = dfs(grid, (y, x), M, N)
                area_sizes.append(area_size)
    
    area_sizes.sort()
    return len(area_sizes), area_sizes

# Input
M, N, K = map(int, input().split())
rectangles = [list(map(int, input().split())) for _ in range(K)]

# Process and Output
count, areas = calculate_areas(M, N, rectangles)
print(count)
print(' '.join(map(str, areas)))

- 개선점

1. 인접리스트로 초기화 할때 영역을 곧바로 디스트럭쳐링 하고 -> 이 값에 대해서 range()로 범위를 잡아서 했음

 

2. 굳이 빈 값 리스트를 잡을 필요없이 size의 갯수를 len으로 해서 쉽게 표현가능

 

 

<보충학습>