글 읽기 - 낚시왕 ..... 처음에 바로 틀렸다고 나옵니다.

iopengom 2년 전 0

테스트 케이스는 다 맞는데

왜 그럴까요..

모르겠습니다.

from collections import deque
R, C, M = map(int, input().split())
Map = [[[] for _ in range(C+1)] for _ in range(R+1)]
shark = []
for i in range(M):
    r,c,s,d,z = map(int, input().split())  # 1부터 시작
    shark.append([r,c,s,d,z])
    Map[r][c].append([i,z])
# 방향 0, 상, 하, 우, 좌
dx, dy = [0, -1, 1, 0, 0], [0, 0, 0, 1, -1]


# 낚시
got_shark = 0
def fishing(col):
    global got_shark
    for i in range(1, R+1):
        if len(Map[i][col]) > 0:
            idx = Map[i][col][0][0]
            got_shark += shark[idx][4]
            Map[i][col] = []
            shark[idx] = []
            break



# 이동
# 1초동안 d의 방향으로 s만큼 이동한다. 벽에 부딪혔을시 반대방향으로간다.
# shark, map 갱신
def move(idx):
    if shark[idx] == []: # 없어진 상어
        return
    x, y, s, d, z = shark[idx]
    nx, ny = x + s*dx[d], y + s*dy[d]
    #nx, ny = nx % (R+(R-2)), ny % (C+(C-2))
    while True:
        if nx < 1 or nx > R:
            if nx < 1:
                nx = 1-(nx)+1
                d = 2
            elif nx > R:
                nx = R - (nx-R)
                d = 1
        elif ny < 1 or ny > C:
            if ny < 1:
                ny = 1-(ny)+1
                d = 3
            elif ny > C:
                ny = C - (ny-C)
                d = 4
        else:
            break
    shark[idx] = [nx, ny, s, d, z]
    Map[x][y] = []
    Map[nx][ny].append([idx,z])

# 취식
def eat():
    for i in range(1, R+1):
        for j in range(1, C+1):
            if len(Map[i][j]) > 1:
                lst = Map[i][j]
                big_s = max(lst, key = lambda x: x[1])
                for s in lst:
                    if s[0] != big_s[0]:
                        shark[s[0]] = [] # 상어 리스트에서 지우기
                Map[i][j] = [big_s] # 제일 큰놈만 남기기



for i in range(1, C+1):
    fishing(i)
    for idx in range(len(shark)):
        move(idx)
    eat()

print(got_shark)

smigyeong 1년 전 0

안녕하세요! 너무 시간이 많이 흘렀는데, 혹시나 해서 남깁니다!

shark[idx] = [nx, ny, s, d, z]
Map[x][y] = []
Map[nx][ny].append([idx,z])

이 부분에서 Map[x][y] = [] 와 같이 이전 위치를 빈 공간 처리하는 부분에서 문제가 생기는 것 같습니다!

이와 관련해서 게시판 글 올렸는데 참고 부탁드려요 :)

https://www.acmicpc.net/board/...

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

문제

문제

출처

ICPC

낚시왕 ..... 처음에 바로 틀렸다고 나옵니다.

iopengom 2년 전 0

smigyeong 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말