글 읽기 - (pypy3) 테스트 케이스를 통과하지 못합니다. 로직에 어떤 문제가 있을까요?

hcjy97 2년 전 0

테스트 케이스 4번을 통과하지 못하고 있습니다.

지금까지 디버깅하면서 확인한 부분은 다음과 같습니다.

- 파이어볼이 2개 이상인 경우 각각의 파이어볼의 방향을 홀/짝으로 구분해서 새로운 방향을 결정하기

- 파이어볼의 질량의 합 / 5가 0이 아닌 경우에만 새로운 파이어볼 만들기

- 새로운 파이어볼을 만들기 전에 칸을 비워주기 (21번째 줄에서 popleft()를 사용해서 기존 파이어볼을 아예 꺼내기 때문에 while문 종료 후에는 칸이 비게 됨)

- 파이어볼을 이동시킬 때 좌표를 결정하는 부분 확인 (1과 n이 이어져 있는 부분)

제가 어떤 로직을 빠트려서 4번 테스트케이스를 통과하지 못하는지 궁금합니다. 정답은 9인데 제 코드로는 14가 나오네요..

import math
from collections import deque

# 입력
n, m, k = map(int, input().split())
fireinfo = [list(map(int, input().split())) for _ in range(m)]
field = [[deque() for _ in range(n)] for _ in range(n)]
for y, x, m, s, d in fireinfo:
    field[y-1][x-1].append([m, s, d])

# 파이어볼이 2개 이상인 칸에서 파이어볼을 4개로 만드는 함수
def merge():
    global n, field
    for y in range(n):
        for x in range(n):
            # 길이가 2 이상인 경우에만
            if len(field[y][x]) > 1:
                cnt = len(field[y][x])
                odd, even, weight, speed = 0, 0, 0, 0
                # deque에 있는 원소를 popleft하면서 odd, even, weight, speed를 채우기
                while field[y][x]:
                    m, s, d = field[y][x].popleft()
                    weight += m
                    speed += s
                    if d % 2 == 0:
                        even += 1
                    elif d % 2 > 0:
                        odd += 1
                # popleft하기 때문에 여기서 field[y][x]는 deque()로 초기화됨

                # 새로운 파이어볼의 weight와 speed 구하기 (버림 - floor)
                nxt_weight = math.floor(weight / 5)
                nxt_speed = math.floor(speed / cnt)

                # weight가 0인 경우 고려
                if nxt_weight != 0:
                    # 방향이 짝수인 파이어볼의 개수와 홀수인 파이어볼의 개수를 고려
                    if odd == 0 or even == 0:
                        for nd in [0, 2, 4, 6]:
                            field[y][x].append([nxt_weight, nxt_speed, nd])
                    else:
                        for nd in [1, 3, 5, 7]:
                            field[y][x].append([nxt_weight, nxt_speed, nd])

dxy = [(-1, 0), (-1, 1), (0, 1), (1, 1), (1, 0), (1, -1), (0, -1), (-1, -1)]

for _ in range(k):
    # 배열을 탐색하여 파이어볼이 있는 위치를 저장
    fireExist = []
    for y in range(n):
        for x in range(n):
            if field[y][x]:
                fireExist.append([y, x])
    # 파이어볼을 움직이기
    for y, x in fireExist:
        while field[y][x]:
            m, s, d = field[y][x].popleft()
            ny, nx = y + dxy[d][0]*s, x + dxy[d][1]*s
            ny, nx = (ny + n) % n, (nx + n) % n
            field[ny][nx].append([m, s, d])
    # 파이어볼이 2개 이상인 경우 4개로 만들기
    merge()

# 남아 있는 파이어볼의 질량의 합을 구하기
res = 0
for y in range(n):
    for x in range(n):
        if field[y][x]:
            for m, s, d in field[y][x]:
                res += m
print(res)

hcjy97 2년 전 0

자문자답입니다.

이유는 알 수 없으나 파이어볼을 움직일 때

for y, x in fireExist:
while field[y][x]:

이런 식으로 움직이는 부분이 문제였습니다.

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

문제

문제

출처

ICPC

(pypy3) 테스트 케이스를 통과하지 못합니다. 로직에 어떤 문제가 있을까요?

hcjy97 2년 전 0

hcjy97 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말