글 읽기 - 모든 반례 다 통과하는데.... 틀렸습니다 뜹니다 제발 도와주세요 ㅠㅠ

kkjjhhbb 1년 전 0

도대체 뭐가 틀린걸까요..

온갖 반례들 다 넣어봤을 때 모두 잘 작동하였습니다.ㅠㅠㅠㅠㅠㅠㅠㅠㅠㅠ

from collections import deque
n=int(input())
arr=[]
for i in range(n):
    inn=list(map(int,input().split()))
    arr.append(inn)
    for j in range(n):
        if inn[j] == 9:
            bx,by=i,j #상어 위치

#북 남 서 동
dx=[-1,1,0,0]
dy=[0,0,-1,1]

sang=[] #상어가 먹을 수 있는 애들 리스트
baby=2 #상어 크기
eat=0 #먹은 횟수
cnt = 0 #몇 초 걸리는 지

def cannoteat(): #종료 확인
    chk=False
    all=0
    for i in range(n):
        for j in range(n):
            if i==bx and j==by:
                continue
            if arr[i][j]<baby:
                chk=True
                all+=arr[i][j]
    if not chk or all == 0:
        return True
    return False

while True:
    q = deque()
    q.append((bx, by))
    visited = [[0] * n for _ in range(n)]
    visited[bx][by]=True
    dist = [[0] * n for _ in range(n)]
    sang = []
    if cannoteat():
        break
    while q:
        x,y=q.popleft()
        if cannoteat():
            break
        turn=0
        for i in range(4):
            nx,ny=x+dx[i],y+dy[i]
            if 0<=nx<n and 0<=ny<n and arr[nx][ny] <= baby:
                if not visited[nx][ny]:
                    visited[nx][ny]=True
                    dist[nx][ny]=dist[x][y]+1
                    q.append((nx,ny))
                    if 1<=arr[nx][ny]<baby:
                        sang.append([nx,ny,dist[nx][ny]])
            else:
                turn+=1
    if turn==4:
        break
    if sang: #먹기 위해서
        sang=sorted(sang,key=lambda x:(x[2],x[0],x[1]),reverse=True) #제일 가까운 것들 중에 제일 위쪽, 왼쪽인거,
        xx,yy,tt=sang.pop()
        cnt+=tt
        eat+=1
        arr[xx][yy]=0
        arr[bx][by]=0
        bx,by=xx,yy
    if eat == baby: #상어의 몸집이 먹은 횟수와 같다면
        eat=0
        baby+=1

print(cnt)

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

문제

문제

출처

ICPC

모든 반례 다 통과하는데.... 틀렸습니다 뜹니다 제발 도와주세요 ㅠㅠ

kkjjhhbb 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말