글 읽기 - 파이썬 원판돌리기 11%에서 자꾸 실패인데...

pyoun820 4년 전 0

11%에서 실패인데, 일단 예제는 기본적으로 다 일치하고 반례들도 넣어보는데 제가 보기엔 문제되는 부분이 안보여서요,,

11%의 테스트케이스의 반례가 무엇이 있을지 아니면, 코드의 잘못된 부분을 지적해주시면 감사합니다.

from collections import deque

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

def avg(arr):
    total, N = 0, 0
    for i in range(row):
        for j in range(col):
            if arr[i][j]:
                total += arr[i][j]
                N += 1

    return total/N if total else 0



def erase():
    flag = 1
    visit = [[0 for _ in range(col)] for _ in range(row)]
    for y in range(row):
        for x in range(col):

            if board[y][x] and not visit[y][x]:
                visit[y][x] = 1
                number = board[y][x]

                dq = deque()
                dq.append([y, x])
                while dq:
                    y, x = dq.popleft()
                    for d in range(4):
                        new_y = y + dy[d]
                        new_x = x + dx[d]
                        if new_x == -1:
                            new_x = col -1
                        elif new_x == col:
                            new_x = 0

                        if 0 <= new_y < row and 0 <= new_x < col and board[new_y][new_x] == number\
                                and not visit[new_y][new_x]:
                            dq.append([new_y, new_x])
                            board[new_y][new_x], board[y][x] = 0, 0
                            visit[new_y][new_x] = 1

                    if flag and len(dq):  # 덱에 하나라도 들어가면 현재 좌표의 값을 0으로하고 리턴 0을 한다.
                        flag = 0

    return avg(board) if flag else 0

row, col, num_of_case = map(int ,input().split())  # 행 크기, 열 크기, 총 케이스 개수

board = []
for r in range(row):   # 보드 생성
    board.append(deque(map(int, (input().split()))))

for case in range(num_of_case):
    num, direct, rotate = map(int, input().split())  # num 의 배수인 원판을 대상으로, d = 0 시계 d = 1 반시계, 회전수
    if direct:  # 반시계 방향
        for n in range(num, row+1, num):
            n -= 1  # 인덱스 맞추기 ex) 2 -> 1 , 4 -> 3
            for _ in range(rotate): board[n].append(board[n].popleft())

    else:  # 시계 방향
        for n in range(num, row+1, num):
            n -= 1  # 인덱스 맞추기 ex) 2 -> 1 , 4 -> 3
            for _ in range(rotate): board[n].appendleft(board[n].pop())

    value = erase()
    if value:
        for p in range(row):
            for q in range(col):
                if board[p][q]:
                    if board[p][q] < value:
                        board[p][q] += 1
                    elif board[p][q] > value:
                        board[p][q] -= 1
answer = 0

for r in range(row):
    answer += sum(board[r])
print(answer)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

파이썬 원판돌리기 11%에서 자꾸 실패인데...

pyoun820 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말