글 읽기 - 파이썬 19프로에서 틀리는데 어디가 문제일까요?

kis03160 2년 전 0

파이썬 19프로에서 틀리는데 어디가 문제일까요?

질문게시판에 있는 반례들 넣었을 때 다 정상적으로 출력하는거 같은데...

힌트나 반례 주시면 감사드리겠습니다!

def remove_dup(P):
    check = [[0 for _ in range(m)] for _ in range(n + 1)]

    for i in range(1, len(P)):
        for j in range(len(P[i])):
            if P[i][j] == 0:
                continue
            value = P[i][j]

            # 이미 지워진 것 -> 이전 값 가져옴
            if value == -1:
                value = check[i][j]

            # 깉은 원판 내 인접
            if value == P[i][(j + 1) % len(P[i])]:
                adj = (j + 1) % len(P[i])
                check[i][j] = check[i][adj] = max(value, check[i][adj])
                P[i][j] = P[i][adj] = -1
            elif value == P[i][j - 1]:
                check[i][j] = check[i][j - 1] = max(value, check[i][j - 1])
                P[i][j] = P[i][j - 1] = -1

            # 서로 다른 원판 간 인접
            if i + 1 < len(P) and value == P[i + 1][j]:
                check[i][j] = check[i + 1][j] = max(value, check[i + 1][j])
                P[i][j] = P[i + 1][j] = -1

            elif i - 1 != 0 and value == P[i - 1][j]:
                check[i][j] = check[i - 1][j] = max(value, check[i - 1][j])
                P[i][j] = P[i - 1][j] = -1


def rotate(P, x, d):
    for i in range(x, len(P), x):
        # 시계
        if d == 0:
            zero = P[i][-1]
            for j in range(len(P[i]) - 1, 0, -1):
                P[i][j] = P[i][j - 1]
            P[i][0] = zero
        # 반시계
        else:
            zero = P[i][0]
            for j in range(len(P[i]) - 1):
                P[i][j] = P[i][j + 1]
            P[i][-1] = zero


def sum_and_check_deleted(P):
    result = 0
    count = 0
    is_deleted = False
    for i in range(1, len(P)):
        for j in range(len(P[i])):
            if P[i][j] > 0:
                count += 1
            if P[i][j] < 0:
                is_deleted = True
                P[i][j] = 0
            result += P[i][j]
    return is_deleted, count, result


def cal(P, result, count):
    avg = result / count
    for i in range(1, len(P)):
        for j in range(len(P[i])):
            if P[i][j] > avg and P[i][j]:
                P[i][j] -= 1
                result -= 1
            elif P[i][j] < avg and P[i][j]:
                P[i][j] += 1
                result += 1
    return result


def answer(P, T):
    result = 0
    count = 0
    for x, d, k in T:
        result = 0

        for _ in range(k):
            rotate(P, x, d)
        remove_dup(P)

        is_deleted, count, result = sum_and_check_deleted(P)

        if count == 0 or is_deleted:
            continue
        result = cal(P, result, count)

    return result if count else 0


import sys

input = sys.stdin.readline
n, m, t = map(int, input().split())
P = [[]] + [list(map(int, input().split())) for _ in range(n)]
T = [list(map(int, input().split())) for _ in range(t)]
print(answer(P, T))

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

문제

문제

출처

ICPC

파이썬 19프로에서 틀리는데 어디가 문제일까요?

kis03160 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말