[BAEKJOON python] 17822_원판 돌리기

원판 돌리기
반지름이 1, 2, ..., N인 원판이 크기가 작아지는 순으로 바닥에 놓여있고
원판의 중심은 모두 같다
원판의 반지름이 i이면, 그 원판을 i번째 원판
각각의 원판에는 M개의 정수가 적혀있고, i번째 원판에 적힌 j번째 수의 위치는 (i, j)

수의 위치
(i, 1)은 (i, 2), (i, M)과 인접
(i, M)은 (i, M-1), (i, 1)과 인접
(i, j)는 (i, j-1), (i, j+1)과 인접 (2 ≤ j ≤ M-1)
(1, j)는 (2, j)와 인접
(N, j)는 (N-1, j)와 인접
(i, j)는 (i-1, j), (i+1, j)와 인접 (2 ≤ i ≤ N-1)

원판의 회전은 독립적으로 이루어진다
2번 원판을 회전했을 때, 나머지 원판은 회전하지 않는다
원판을 회전시킬 때는 수의 위치를 기준으로 하며, 회전시킨 후의 수의 위치는 회전시키기 전과 일치

원판을 아래와 같은 방법으로 총 T번 회전
원판의 회전 방법은 미리 정해져 있고, i번째 회전할때 사용하는 변수는 xi, di, ki
1) 번호가 xi의 배수인 원판을 di방향으로 ki칸 회전
    0인 경우는 시계 방향, 1인 경우는 반시계 방향
2) 원판에 수가 남아 있으면, 인접하면서 수가 같은 것을 모두 찾는다
    그러한 수가 있는 경우에는 원판에서 인접하면서 같은 수를 모두 지운다
    없는 경우에는 원판에 적힌 수의 평균을 구하고, 평균보다 큰 수에서 1을 빼고, 작은 수에는 1을 더한다
3) 원판을 T번 회전시킨 후 원판에 적힌 수의 합

첫째 줄에 N, M, T이 주어진다
둘째 줄부터 N개의 줄에 원판에 적힌 수가 주어진다
    i번째 줄의 j번째 수는 (i, j)에 적힌 수를 의미
다음 T개의 줄에 xi, di, ki

from collections import deque

# n개의 원판, 원판에는 M개의 정수, t번 회전
n, m, t = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(n)]
# 동, 서, 남, 북
dx = [0, 0, -1, 1]
dy = [1, -1, 0, 0]

def rotate(x, d, k) :
    queue = deque()
    queue.extend(graph[x])
    # 0인 경우는 시계 방향, 1인 경우는 반시계 방향
    if d == 0 :
        # 시계방향으로 k배 만큼 돌리기
        queue.rotate(k)
    else :
        # 반시계방향으로 k배 만큼 돌리기
        queue.rotate(-k)
    graph[x] = list(queue)

def change_arg() :
    arg_cnt = 0
    all_sum = 0
    for i in range(n) :
        for j in range(m) :
            # 처음 모든 제거된 숫자(0)을 제외한 나머지 숫자의 갯수와 
            # 그 숫자들의 합을 계산한 후,
            # 만약 모든 숫자가 0이라면 평균을 계산하지 않게 리턴
            if graph[i][j] != 0 :
                arg_cnt += 1
                all_sum += graph[i][j]
            
    if arg_cnt == 0:
        return False
    # 평균을 계산했다면 모든 원판에 들어있는 숫자들에 값과 비교
    avg = all_sum / arg_cnt
    for i in range(n) :
        for j in range(m) :
            if graph[i][j] != 0 :
                # 평균 값보다 크다면 -1, 평균값보다 작다면 +1
                if graph[i][j] > avg :
                    graph[i][j] -= 1
                elif graph[i][j] < avg :
                    graph[i][j] += 1
    return True

# 인접한 곳에 숫자가 같은지 파악해서 만약 숫자가 같다면 원판에 숫자를 제거(0으로 변경)
# i번째 원판에는 즉, 0번 열과 m-1 열은 이어져 있으므로 인접 4방향을 살필 때, 값을 잘 변경해줘서 모든 원판에 인접한 곳을 방문
def solve(x, y) :
    queue = deque()
    queue.append((x, y))
    visited[x][y] = True
    
    value = graph[x][y]
    graph[x][y] = 0
    cnt = 0
    while queue :
        x, y = queue.popleft()
        for i in range(4) :
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 > ny or ny >= m :
                if y == 0 :
                    ny = m -1
                elif y == m - 1:
                    ny = 0
            if 0 <= nx < n and 0 <= ny < m :
                if not visited[nx][ny] :
                    if graph[nx][ny] == value :
                        cnt += 1
                        graph[nx][ny] = 0
                        visited[nx][ny] = True
                        queue.append((nx, ny))
    if cnt == 0 :
        graph[x][y] = value
    return cnt

for _ in range(t) :
    # 번호가 xi의 배수인 원판을 di방향으로 ki칸 회전
    x, d, k = map(int, input().split())
    check_sum = 0
    for i in range(n) :
        check_sum += sum(graph[i])
        if (i+1) % x == 0 :
            rotate(i, d, k)
    if check_sum == 0 :
        break
    else :
        visited = [[False] * m for _ in range(n)]
        cnt = 0
        for i in range(n) :
            for j in range(m) :
                if not visited[i][j] and graph[i][j] != 0 :
                    cnt += solve(i, j)
        if cnt == 0 :
            change_arg()
answer = 0
for i in range(n) :
    answer += sum(graph[i])
print(answer)

---

예제 입력

4 4 1
1 1 2 3
5 2 4 2
3 1 3 5
2 1 3 2
2 0 1

예제 출력

30

 

 

 

쓰앵님 어렵습니다.

https://developer-ellen.tistory.com/65