글 읽기 - 게시판 모든 반례와 예시 다 패스하는데 4%에서 틀립니다.

qorgod00 2년 전 0

게시판 모든 반례와 예시 다 패스하는데 4%에서 틀립니다. 틀린부분 알려주심 감사하겠습니다.

import math
from collections import deque

N, L, R = map(int, input().split())
A = [list(map(int, input().split())) for _ in range(N)]

# N, L, R = 2, 20 ,50
# A = [[50, 30], [20, 40]]

# N, L, R = 3, 5 ,50
# A = [[50, 30, 20], [20, 40, 30], [20, 20, 30]]

# N, L, R = 3, 5 ,50
# A = [[50, 40, 50], [50, 50, 50], [50, 50, 50]]

# N, L, R = 3, 5 ,50
# A = [[50, 40, 50], [50, 50, 50], [50, 50, 50]]

# N, L, R = 4, 1, 9
# A = [[96, 93, 74, 30], [60, 90, 65, 96], [5, 27, 17, 98], [10, 41, 46, 20]]

def update_A(tracked, A):
    for team, ls in tracked.items():
        if len(ls) < 1:
            continue
            
        _sum = 0
        for (x, y) in ls:
            _sum += A[x][y]
        _avg = math.trunc(_sum/len(ls))
        
        for (x, y) in ls:
            A[x][y] = _avg   
    return A
        
def iter_population(A):
    global N, L, R
    team_dict = dict(); visited = set(); tracked = dict() # {1:[(), (), (), ...], 2: ...}
    q = deque([])
    flag = False
    for i in range(N):
        for j in range(N):
            q.append((i, j))
            team_dict[(i, j)] = -1
            
    team_n = 0
    while q:
        x, y = q.popleft()
        team = team_dict[(x, y)]
        if team == -1:
            team = team_n
            team_dict[(x, y)] = team_n
            tracked[team_n] = set()
            tracked[team_n].add((x, y))
            team_n += 1
            
        for dx, dy in zip([0,0,1,-1], [1,-1,0,0]):
            nx, ny = x+dx, y+dy
            if 0 <= nx < N and 0 <= ny < N and team_dict[(nx, ny)] == -1:
                if L <= abs(A[nx][ny]-A[x][y]) <= R:
                    tracked[team].add((nx, ny))
                    visited.add((nx, ny))
                    team_dict[(nx, ny)] = team
                    q.append((nx, ny))
                    flag = True
    if flag:
        A = update_A(tracked, A)
        return A
    else:
        return None
    
fin = False
ans = 0
while not fin:
    A_prime = iter_population(A)
    if A_prime is not None:
        A = A_prime
        ans += 1
    else:
        fin = True
print(ans)

seobank1 11달 전 0

혹시 해결하셨습니까... 저도 4%에서 틀려서 ㅠ

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

문제

문제

출처

ICPC

게시판 모든 반례와 예시 다 패스하는데 4%에서 틀립니다.

qorgod00 2년 전 0

seobank1 11달 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말