글 읽기 - python에서 바로 틀렸습니다 뜨고 반례도 다 맞는데 뭐가 문제일까요ㅜㅜㅜㅠㅠ

ektha3869 3년 전 0

ㅜㅜpython에서 바로 틀렸습니다 뜨고 반례도 다 맞는데 뭐가 문제일까요ㅜㅜㅜㅠㅠ

# 1: 위, 2: 아래, 3: 오른쪽, 4: 왼쪽
dB = [0, -1, 1, 1, -1]

# 상어 이동하는거 + 상어 잡아먹히는거
def shark():
    global board
    tmp = [[0] * C for _ in range(R)]
    for i in range(R):
        for j in range(C):
            # 0. 상어 발견
            if board[i][j] != 0:
                # 1. 이동하는 위치 i / j / 방향 / 이동횟수
                si, sj, di, cnt = i, j, board[i][j][1], 0
                # 2. 속력이 0이 아닐때 속력만큼 상어 이동
                if board[i][j][0] != 0:
                    while cnt < board[i][j][0]:
                        # 2-1. 방향이 위, 아래일때
                        if di == 1 or di == 2:
                            if si == 0:
                                if di == 1:
                                    di = 2
                            elif si == R - 1:
                                if di == 2:
                                    di = 1
                            si += dB[di]
                        # 2-2. 방향이 오른쪽, 왼쪽일때
                        else:
                            if sj == 0:
                                if di == 4:
                                    di = 3
                            elif sj == C - 1:
                                if di == 3:
                                    di = 4
                            sj += dB[di]
                        cnt += 1
                # 3. 상어 크기 비교해서 tmp에 배치하기
                if tmp[si][sj] != 0:
                    if board[i][j][2] > tmp[si][sj][2]:
                        tmp[si][sj] = [board[i][j][0], di, board[i][j][2]]
                else:
                    tmp[si][sj] = [board[i][j][0], di, board[i][j][2]]
    # 4. tmp를 board로 변경
    board = tmp


R, C, M = map(int, input().split())
board = [[0] * C for _ in range(R)]
size = 0
if M != 0:
    # 0. board에 상어 배치
    for _ in range(M):
        r, c, s, d, z = map(int, input().split())
        board[r - 1][c - 1] = [s, d, z]

    # 1. 사람 이동 + 상어 냠냠
    for j in range(C):
        for i in range(R):
            # 2. 이동한 곳에 상어 있으면 냠냠 + 상어 이동
            if board[i][j] != 0:
                size += board[i][j][2]
                board[i][j] = 0
                shark()
                break
        else:
            shark()

print(size)

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

문제

문제

출처

ICPC

python에서 바로 틀렸습니다 뜨고 반례도 다 맞는데 뭐가 문제일까요ㅜㅜㅜㅠㅠ

ektha3869 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말