18-SeongHoonC

[SeongHoonC]

🔗 문제 링크

인구 이동

✔️ 소요된 시간

1시간

✨ 수도 코드

1. 인구가 더 이상 이동할 수 없을 때까지 반복하면서 카운트 증가
2. 인구를 bfs 로 연합 찾기 (L 이상 R 이하여야한다)
3. 찾은 연합 리스트 아이템들을 다 평균값으로 바꾼다.
4. 만약 연합을 하나도 못찾으면 인구를 더 이상 이동시킬 수 없다 -> 종료

📚 새롭게 알게된 내용

[alstjr7437]

저도 똑같은 알고리즘으로 풀었는데
성훈님처럼 함수화를 따로 하지는 않아서 조금 보기 힘들어서 주석 추가했습니다!!

from collections import deque
import sys

def BFS(y,x) :
    queue = deque()
    result = []
    queue.append((x,y))
    result.append((x,y))
    while queue : 
        x,y = queue.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < N and 0 <= ny < N and visited[ny][nx] == 0:
                if L <= abs(ground[y][x] - ground[ny][nx]) <= R:     # L,R로 동맹 찾기
                    visited[ny][nx] = 1
                    queue.append((nx,ny))
                    result.append((nx,ny))
    return result

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

N,L,R = map(int, input().split())

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

day = 0 

while 1:
    visited = [[0] * (N+1) for _ in range(N+1)]
    stop = 0
    for i in range(N):
        for j in range(N):
            if visited[i][j] == 0:
                visited[i][j] = 1
                result = BFS(i,j)       # 동맹을 맺은 나라 가져오기
                if len(result) > 1:     # 동맹이 있으면 평균값 넣기
                    stop = 1
                    num = sum([ground[y][x] for x, y in result]) // len(result)
                    for x,y in result:
                        ground[y][x] = num
    if stop == 0:       # 동맹을 맺을 수 없다면
        break
    day += 1

print(day)

[alstjr7437]

평균값까지 함수화하셔서 만드셨군요!

[wkdghdwns199]

BFS 연습용으로 좋은 문제 감사함다! 보면서 풀었지만..! 😨

import sys
from collections import deque
input = sys.stdin.readline
N,L,R = map(int, input().split())
country = [list(map(int, input().split())) for _ in range(N)]
q = deque()
#연합이 될 수 있을지 확인 후 연랍이 되면 저장
dx = [1,0,-1,0]
dy = [0,1,0,-1]

def bfs(x,y):
    q.append((x,y))
    union=[]
    union.append((x,y))
    while q:
        a,b = q.popleft()
        for i in range(4):
            na = a + dx[i]
            nb = b + dy[i]
            if na >= N or nb>= N or nb<0 or na <0 or visited[na][nb]==1:
                continue
            if R>=abs(country[a][b]-country[na][nb]) >= L:
                visited[na][nb] = 1
                q.append((na,nb))
                union.append((na,nb))
    if len(union)<=1:
        return 0
    result=sum(country[a][b] for a,b in union) // len(union)
    for a,b in union:
        country[a][b] = result

    return 1
day=0

while True :
    stop = 0
    visited = [[0]*N for _ in range(N)]
    for i in range(N):
        for j in range(N):
            if visited[i][j] == 0:
                visited[i][j] = 1
                stop += bfs(i,j)
    if stop ==0:
        break
    day+=1
print(day)

[fnzksxl]

풀고보니까 민석님이랑 거의 똑같이 풀리네요.. ㅋㅋㅋ
재밌는 문제네요! 파이썬의 느린 속도가 크게 와닿는 문제였어요..

from collections import deque
import sys

input = sys.stdin.readline

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

def bfs(x, y):
    union = []
    q = deque()
    
    union.append((x,y))
    q.append((x,y))

    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0 <= nx < N and 0 <= ny < N and not visited[nx][ny] and L <= abs(_map[x][y] - _map[nx][ny]) <= R:
                    visited[nx][ny] = True
                    q.append((nx,ny))
                    union.append((nx,ny))
    
    return union


N, L, R = map(int, input().split())

_map = []

for _ in range(N):
    _map.append(list(map(int, input().split())))

answer = 0

while True:
    flag = True
    visited = [[False] * N for _ in range(N)]
    for i in range(N):
        for j in range(N):
            if not visited[i][j]:
                visited[i][j] = True
                union = bfs(i,j)
                if len(union) > 1:
                    flag = False
                    human = 0
                    for x, y in union:
                        human += _map[x][y] 
                    human //= len(union)

                    for x, y in union:
                        _map[x][y] = human

    if flag:
        break
    
    answer += 1

print(answer)