[백준 16234] 인구이동

문제

NxN 크기의 땅이 있고 1X1개의 칸으로 나누어져있다. 각각의 땅에는 나라가 존재하고 몇 명이 사는지 알 수 있다.

인구이동이 시작할 때는 다음과 같이 진행되고 인구 이동을 할 수 없을 때까지 반복한다.

• 국경선을 공유하는 두 나라의 인구 차이가 L명 이상, R명 이하라면, 두 나라가 공유하는 국경선을 오늘 하루동안 연다.
• 위의 조건에 의해 열어야하는 국경선이 모두 열렸다면, 인구 이동을 시작한다.
• 국경선이 열려있어 인접한 칸만을 이용해 이동할 수 있으면, 그 나라를 오늘 하루 동안은 연합이라고 한다.
• 연합을 이루고 있는 각 칸의 인구수는 (연합의 인구수) / (연합을 이루고 있는 칸의 개수)가 된다. 편의상 소수점은 버린다.
• 연합을 해체하고, 모든 국경선을 닫는다.

각 나라의 인구수가 주어졌을 때 인구 이동이 몇 번 발생하는지 구하라.

 

입력

첫째 줄에 N, L, R이 주어진다. (1 ≤ N ≤ 50, 1 ≤ L ≤ R ≤ 100)

둘째 줄부터 N개의 줄에 각 나라의 인구수가 주어진다. r행 c열에 주어지는 정수는 A[r][c]의 값이다. (0 ≤ A[r][c] ≤ 100)

인구 이동이 발생하는 횟수가 2,000번 보다 작거나 같은 입력만 주어진다.

 

출력

인구 이동이 몇 번 발생하는지 첫째 줄에 출력한다.

---

접근

- 이중 for문을 탐색하여 특정 좌표에서 인구이동을 할 수 있는지 체크한다.

- 인구이동을 할 수 있다면 BFS로 탐색하여 평균과 국경선을 공유하는 좌표를 반환한다.

- 이중 for문이 종료되었다면 인구이동을 했던 좌표들에 평균값을 대입한다.

- 인구이동이 일어났다면 최종해를 +1한다.

 

구현

- is_update가 False이면 반복문을 종료한다.

- final_visited에는 국경선을 개방한 모든 나라의 좌표를 넣어둔다.

- values에는 평균값과 해당 평균값을 넣을 나라의 좌표를 튜플형태로 넣는다.

- 이중 for문을 탐색할 때 인구이동이 일어날 수 있다면 bfs를 시작하고 평균과 해당 좌표를 반환받아 values에 넣는다.

- 그리고 final_visited에 위에서 인구이동이 일어나는 나라의 좌표를 합친다.

- 이중 for문이 종료되고 values에 해당하는 값을 각 나라에 평균값을 넣는다.

- 만약 is_update가 True면 최종해를 +1한다.

check = True
is_update = True
answer = 0
while is_update:
    is_update = False
    final_visited = set()
    values = []
    for y in range(N):
        for x in range(N):
            if is_move(y, x, final_visited):
                avg, visited = bfs(y, x)
                is_update = True

                values.append((avg, visited))
                final_visited |= visited

    for avg, pos in values:
        for y, x in pos:
            Map[y][x] = avg

    if is_update:
        answer += 1
print(answer)

 

- bfs함수에서 국경선을 개방하는 나라를 찾는다.

- 그리고 평균과 해당 좌표를 반환한다.

def bfs(y, x):
    q = deque()
    visited = set()
    q.append((y, x))
    visited.add((y, x))
    total = 0

    dy = [1, -1, 0, 0]
    dx = [0, 0, 1, -1]
    while q:
        cy , cx = q.popleft()
        total += Map[cy][cx]

        for d in range(4):
            ny = cy + dy[d]
            nx = cx + dx[d]
            if check_range(ny, nx):
                if L <= abs(Map[ny][nx] - Map[cy][cx]) <= R and (ny, nx) not in visited:
                    visited.add((ny, nx))
                    q.append((ny, nx))


    avg = total // len(visited)
    return avg, visited

---

전체 코드

import sys
from _collections import deque
input = sys.stdin.readline

def check_range(y, x):
    return (0 <= y < N) and (0 <= x < N)

def is_move(y, x, visited):
    dy = [1, -1, 0, 0]
    dx = [0, 0, 1, -1]

    for d in range(4):
        ny = y + dy[d]
        nx = x + dx[d]
        if check_range(ny, nx):
            if L <= abs(Map[ny][nx] - Map[y][x]) <= R and (ny, nx) not in visited:
                return True
    return False

def bfs(y, x):
    q = deque()
    visited = set()
    q.append((y, x))
    visited.add((y, x))
    total = 0

    dy = [1, -1, 0, 0]
    dx = [0, 0, 1, -1]
    while q:
        cy , cx = q.popleft()
        total += Map[cy][cx]

        for d in range(4):
            ny = cy + dy[d]
            nx = cx + dx[d]
            if check_range(ny, nx):
                if L <= abs(Map[ny][nx] - Map[cy][cx]) <= R and (ny, nx) not in visited:
                    visited.add((ny, nx))
                    q.append((ny, nx))


    avg = total // len(visited)
    return avg, visited

N, L, R = map(int, input().split())
Map = [list(map(int, input().split())) for _ in range(N)]

check = True
is_update = True
answer = 0
while is_update:
    is_update = False
    final_visited = set()
    values = []
    for y in range(N):
        for x in range(N):
            if is_move(y, x, final_visited):
                avg, visited = bfs(y, x)
                is_update = True

                values.append((avg, visited))
                final_visited |= visited

    for avg, pos in values:
        for y, x in pos:
            Map[y][x] = avg

    if is_update:
        answer += 1
print(answer)