글 읽기 - [파이썬 주석] 제 풀이에 틀린점을 도저히 찾지 못하겠습니다.

raiders032 3년 전 0

테스트 케이스를 모두 통과하고 제출하자마자 틀렸습니다.

문제의 조건과 제 코드를 대조하면 어떤 부분을 놓쳤는지 열심히 찾아봤는데 못찾았습니다.

도와주세요

N, M, K = map(int, input().split())
# 격자판의 위치(x,y)와 그 위에 있는 파이이어볼들의 질량(m), 속도(s), 방향(d)을 나태냄 {key: (x,y) value:[(m,s,d)...]}
board = dict()
dx = [-1, -1, 0, 1, 1, 1, 0, -1]
dy = [0, 1, 1, 1, 0, -1, -1, -1]
res = 0


def move():
    global board
    # 이동을 위한 임시 딕션너리
    tmp_board = dict()
    for (x, y), msd in board.items():
        for i in range(len(msd)):
            # 파이어볼 다음 이동 위치 구하기
            nx = x + dx[msd[i][2]] * msd[i][1]
            ny = y + dy[msd[i][2]] * msd[i][1]
            if nx >= N:
                nx %= N
            if ny >= N:
                ny %= N
            if nx < 0:
                nx = N + (nx % -N)
            if ny < 0:
                ny = N + (ny % -N)
            # 파이어볼 이동시키기
            if (nx, ny) in tmp_board:
                tmp_board[(nx, ny)].append((msd[i][0], msd[i][1], msd[i][2]))
            else:
                tmp_board[(nx, ny)] = [(msd[i][0], msd[i][1], msd[i][2])]
    # 파이어볼 위치 갱신
    board = tmp_board


def merge_and_conflict():
    global board
    for (x, y), msd in board.items():
        length = len(msd)
        # 충돌이 없는 경우
        if length < 2:
            continue
        sum_s = 0
        sum_m = 0
        arr_d = []
        for m, s, d in msd:
            sum_m += m
            sum_s += s
            arr_d.append(d)
        board[(x, y)] = []
        # 질량이 0인 파이볼인 경우
        if sum_m // 5 == 0:
            continue
        # 모두 홀수이거나 모두 짝수인 경우
        if len(set(map(lambda v: v % 2, arr_d))) == 1:
            for i in range(0, 8, 2):
                board[(x, y)].append((sum_m//5, sum_s//length, i))
        # 그렇지 않은 경우
        else:
            for i in range(1, 8, 2):
                board[(x, y)].append((sum_m//5, sum_s//length, i))


for _ in range(M):
    x, y, m, s, d = map(int, input().split())
    board[(x-1, y-1)] = [(m, s, d)]

for _ in range(K):
    move()
    merge_and_conflict()

for msd in board.values():
    for m, s, d in msd:
        res += m
print(res)

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

문제

문제

출처

ICPC

[파이썬 주석] 제 풀이에 틀린점을 도저히 찾지 못하겠습니다.

raiders032 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말