7.py

fromcollectionsimportdeque

# 땅의 크기(N), L, R 값을 입력받기
n, l, r=map(int, input().split())

# 전체 나라의 정보(N x N)를 입력 받기
graph= []
for_inrange(n):
graph.append(list(map(int, input().split())))

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

# 특정 위치에서 출발하여 모든 연합을 체크한 뒤에 데이터 갱신
defprocess(x, y, index):
# (x, y)의 위치와 연결된 나라(연합) 정보를 담는 리스트
united= []
united.append((x, y))
# 너비 우선 탐색 (BFS)을 위한 큐 라이브러리 사용
q=deque()
q.append((x, y))
union[x][y] =index# 현재 연합의 번호 할당
summary=graph[x][y] # 현재 연합의 전체 인구 수
count=1# 현재 연합의 국가 수
# 큐가 빌 때까지 반복(BFS)
whileq:
x, y=q.popleft()
# 현재 위치에서 4가지 방향을 확인하며
foriinrange(4):
nx=x+dx[i]
ny=y+dy[i]
# 바로 옆에 있는 나라를 확인하여
if0<=nx<nand0<=ny<nandunion[nx][ny] ==-1:
# 옆에 있는 나라와 인구 차이가 L명 이상, R명 이하라면
ifl<=abs(graph[nx][ny] -graph[x][y]) <=r:
q.append((nx, ny))
# 연합에 추가하기
union[nx][ny] =index
summary+=graph[nx][ny]
count+=1
united.append((nx, ny))
# 연합 국가끼리 인구를 분배
fori, jinunited:
graph[i][j] =summary//count

total_count=0

# 더 이상 인구 이동을 할 수 없을 때까지 반복
whileTrue:
union= [[-1] *nfor_inrange(n)]
index=0
foriinrange(n):
forjinrange(n):
ifunion[i][j] ==-1: # 해당 나라가 아직 처리되지 않았다면
process(i, j, index)
index+=1
# 모든 인구 이동이 끝난 경우
ifindex==n*n:
break
total_count+=1

# 인구 이동 횟수 출력
print(total_count)