글 읽기 - 채점도중 17%에서 런타임에러가 나요!! 도와주세요 고수분들!!

jsh0508 3년 전 0

테스트 케이스 다 맞습니다.
하지만 채점 도중에 17%에서 런타임 에러가 발생합니다.
도저히 런타임 에러가 나는 이유를 찾지 못하겠습니다.
꼭 좀 도와주세요 감사합니다.
class Shark:
    def __init__(self, r, c, s, d, z):
        self.pos = (r, c)
        self.velocity = s
        self.direction = d
        self.size = z

    def move(self, r, c):
        if self.direction == 3:  # 오른쪽 방향일 때
            if self.velocity - (c-self.pos[1]) > 0:  # 현재 위치부터 가장 오른쪽 열까지 남은 칸 수 보다 속도가 클 경우
                quotient = (self.velocity - (c - self.pos[1]) - 1) // (c - 1)   # -1 을 해야하는 중요한 이유!! 몫이 홀수 인지 짝수 인지 제대로 구별해야하기 때문에
                remainder = (self.velocity - (c - self.pos[1]) - 1) % (c - 1) + 1
                if quotient % 2 == 0:  # 몫이 짝수인 경우에는 방향도 바뀌고 위치는 컬럼 - 나머지 이다.
                    self.direction = 4
                    self.pos = (self.pos[0], c - remainder)
                else:
                    self.pos = (self.pos[0], 1 + remainder)
            else:
                self.pos = (self.pos[0], self.pos[1] + self.velocity)
        elif self.direction == 4:  # 왼쪽 방향일 때
            if self.velocity - (self.pos[0] - 1) > 0:  # 현재 위치부터 가장 왼쪽 열까지 남은 칸 수보다 속도가 클 경우
                quotient = (self.velocity - (self.pos[1] - 1) - 1) // (c - 1)
                remainder = (self.velocity - (self.pos[1] - 1) - 1) % (c - 1) + 1
                if quotient % 2 == 0:
                    self.direction = 3
                    self.pos = (self.pos[0], 1 + remainder)
                else:
                    self.pos = (self.pos[0], c - remainder)
            else:
                self.pos = (self.pos[0], self.pos[1] - self.velocity)
        elif self.direction == 1:  # 위쪽 방향일 때
            if self.velocity - (self.pos[0] - 1) > 0:
                quotient = (self.velocity - (self.pos[0] - 1) - 1) // (r - 1)
                remainder = (self.velocity - (self.pos[0] - 1) - 1) % (r - 1) + 1
                if quotient % 2 == 0:
                    self.direction = 2
                    self.pos = (1 + remainder, self.pos[1])
                else:
                    self.pos = (r - remainder, self.pos[1])
            else:
                self.pos = (self.pos[0] - self.velocity, self.pos[1])
        else:  # 아래쪽 방향일 때
            if self.velocity - (r - self.pos[0]) > 0:
                quotient = (self.velocity - (r - self.pos[0]) - 1) // (r - 1)
                remainder = (self.velocity - (r - self.pos[0]) - 1) % (r - 1) + 1
                if quotient % 2 == 0:
                    self.direction = 1
                    self.pos = (r - remainder, self.pos[1])
                else:
                    self.pos = (1 + remainder, self.pos[1])
            else:
                self.pos = (self.pos[0] + self.velocity, self.pos[1])
        return self.pos

class FishKing:
    def __init__(self):
        self.pos = 0
        self.total = 0

    def catch_shark(self, management):
        self.pos += 1
        col = list(zip(*management.shark_info))[self.pos]  # 파이썬에서 2차원 배열의 열 뽑아내기
        if 1 in col:
            r = col.index(1)  # 제일 위에 있는 상어 위치 찾기
            prey = management.find_shark[(r, self.pos)]
            if len(management.shark_list) >= 1:
                management.shark_list.remove(prey)
                self.total += prey.size


class Manage:
    def __init__(self, r, c, m):
        self.R = r
        self.C = c
        self.shark_info = [[0] * (c+1) for _ in range(r+1)]
        self.shark_list = []
        self.find_shark = dict()
        for i in range(m):
            r, c, s, d, z = map(int, input().split())
            shark = Shark(r, c, s, d, z)
            self.find_shark[(r, c)] = shark
            self.shark_info[r][c] = 1
            self.shark_list.append(shark)

    def store_shark_info(self):
        for i in range(len(self.shark_list)):
            self.shark_info[self.shark_list[i].pos[0]][self.shark_list[i].pos[1]] = 1

    def total_shark_movement(self):
        new_shark_info = [[0] * (self.C + 1) for _ in range(self.R + 1)]
        new_shark_list = self.shark_list.copy()
        self.find_shark = dict()
        for i in range(len(new_shark_list)):
            new_shark = self.shark_list[i]
            new_pos = new_shark.move(self.R, self.C)
            if new_shark_info[new_pos[0]][new_pos[1]] == 1:  # 상어가 움직여서 좌표가 겹치는 경우
                temp_shark = self.find_shark[new_pos]
                prev_size = temp_shark.size
                if prev_size > new_shark.size:  # 기존 상어의 크기가 새롭게 움직여서 온 상어 크기보다 큰 경우
                    if len(new_shark_list) >= 1:
                        new_shark_list.remove(new_shark)
                else:
                    if len(new_shark_list) >= 1:
                        new_shark_list.remove(temp_shark)
                    self.find_shark[new_pos] = new_shark  # 기존 상어를 지우고 새로운 상어 정보를 입력해줘야함
            else:
                self.find_shark[new_pos] = new_shark
                new_shark_info[new_pos[0]][new_pos[1]] = 1

        self.shark_info = new_shark_info
        self.shark_list = new_shark_list


R, C, M = map(int, input().split())
manage = Manage(R, C, M)
fish_king = FishKing()

if M != 0:
    for _ in range(C):
        fish_king.catch_shark(manage)
        manage.total_shark_movement()

print(fish_king.total)
댓글을 작성하려면 로그인해야 합니다.
문제

문제

출처

ICPC

채점도중 17%에서 런타임에러가 나요!! 도와주세요 고수분들!!

jsh0508 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말
