글 읽기 - 어디가 틀린걸까요..ㅠㅠ

suy427 5년 전 0

같은 로직으로 bfs, dfs 다 해봤는데 전부 테스트케이스를 통과하는데도

틀렸다고 나옵니다.. 도대체 어디가 틀린걸까요..?ㅠㅠ

import collections

N, L, R = list(map(int, input().split(' ')))

world = []
visited = []
stack = []
queue = collections.deque([])
total = 0

dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]

for _ in range(N):
    world.append(list(map(int, input().split(' '))))


def union(row, col):
    global total, world, visited, queue, stack
    visited[row][col] = 1
    queue.append((row, col))
    isPossible = False

    while(len(queue) > 0):
        cur = queue.popleft()
        visited[cur[0]][cur[1]] = 2
        stack.append((cur[0], cur[1]))
        total += world[cur[0]][cur[1]]

        for next in range(4):
            nextRow, nextCol = cur[0] + dy[next], cur[1] + dx[next]
            if (0 <= nextRow < N) and \
               (0 <= nextCol < N) and \
               visited[nextRow][nextCol] == 0 and \
               (L <= abs(world[nextRow][nextCol] - world[cur[0]][cur[1]]) <= R):

                visited[nextRow][nextCol] = 1
                queue.append((nextRow, nextCol))
                isPossible = True

    return isPossible


def calc():
    change = total // len(stack)

    for position in stack:
        world[position[0]][position[1]] = change


def shift():
    global stack, visited, total
    ret = 0

    for _ in range(2000):
        finished = True
        visited = [[0 for _ in range(N)] for _ in range(N)]

        for row in range(N):
            for col in range(N):
                stack = []
                total = 0
                if union(row, col) is True:
                    finished = False
                    calc()

        if finished is True:
            return ret

        ret += 1


print(shift())

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

문제

문제

출처

ICPC

어디가 틀린걸까요..ㅠㅠ

suy427 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말