글 읽기 - [python3]게시판에서 확인한 반례는 다 통과했는데 틀렸다고 나옵니다ㅠ

ihs3583 4년 전 0

게시판에서 본 테스트케이스는 다 통과하는거같은데 어디에서 잘못됐는지 모르겠습니다

코드 진행은 다음과 같습니다.

최초 입력받은 데이터에서 빙산이 있는 좌표와 높이를 리스트(이하 빙산좌표리스트)에 저장

년차 t를 for문으로 증가시키면서 bfs로 빙산이 2개 이상 나눠지는지 확인 - 2개 이상일 때 flag = True 로 바꾼뒤 그때의 t값을 result에 저장 후 출력

flag가 False라면 저장한 빙산의 좌표에서 4방향으로 0이 있는지 확인 후 0의 개수에 따라 빙산좌표리스트에 빙산의 높이 수정

수정된 빙산좌표리스트를 입력데이터에 적용

t += 1

from collections import deque
# 우, 하, 좌, 상
dx = [1,0,-1,0]
dy = [0,1,0,-1]

def ispass(y,x):
    return 0<=y<N and 0<=x<M and data[y][x] and not visited[y][x]

def bfs(sy,sx):
    global cnt, flag
    que = deque()
    visited[sy][sx] = cnt
    if cnt > 1:
        flag = True
        return
    que.append((sy,sx))
    while que:
        y,x = que.popleft()
        for i in range(4):
            ny,nx = y+dy[i],x+dx[i]
            if ispass(ny,nx):
                visited[ny][nx] = cnt

                que.append((ny,nx))
    cnt += 1



N, M = map(int, input().split())
data = [list(map(int, input().split())) for _ in range(N)]
iceberg = []


icemax = 0
for i in range(N):
    for j in range(M):
        if data[i][j]:
            iceberg.append([i,j,data[i][j]])
        if data[i][j] > icemax:
            icemax = data[i][j]

result = 0

for t in range(icemax+1):
    visited = [[0] * M for _ in range(N)]
    cnt = 1
    flag = False
    for i in range(len(iceberg)):
        if ispass(iceberg[i][0],iceberg[i][1]):
            bfs(iceberg[i][0],iceberg[i][1])
            if flag:
                break
    if flag:
        result = t
        break

    if not flag:
        for i in range(len(iceberg)):
            ice = iceberg[i][2]
            water = 0
            y = iceberg[i][0]
            x = iceberg[i][1]
            for j in range(4):
                ny,nx = y+dy[j],x+dx[j]
                if data[ny][nx] == 0:
                    water += 1
            if ice > water:
                ice -= water
            else:
                ice = 0
            iceberg[i][2] = ice
        for i in range(len(iceberg)):
            y = iceberg[i][0]
            x = iceberg[i][1]
            data[y][x] = iceberg[i][2]

print(result)

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

문제

문제

출처

ICPC

[python3]게시판에서 확인한 반례는 다 통과했는데 틀렸다고 나옵니다ㅠ

ihs3583 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말