[백준 2667] 파이썬 - 단지번호

https://www.acmicpc.net/problem/2667

2667번: 단지번호붙이기

 

 

1. 코드 설명

 

from collections import deque

n=int(input())

arr = [list(map(int, input())) for _ in range(n)]

done =[[0] * n for _ in range(n)]

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

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

def bfs (x,y,cnt) :

  

  q = deque()

  q.append((x,y))

  done[x][y] = cnt

  while q :

    x,y = q.popleft()

    

    for i in range(4):

      nx = dx[i] + x

      ny = dy[i] + y

      if 0 <= nx < n and 0 <= ny < n :

        if arr[nx][ny] ==1 and done[nx][ny] ==0 :

          q.append((nx,ny))

          done[nx][ny] = cnt

cnt = 0

for i in range(n) :

  for j in range(n) :

    if arr[i][j] ==1 and done[i][j] ==0 :

      cnt += 1

      bfs(i,j,cnt)

print (cnt)

 

 

# 2차원인 done 을 1차원으로 변환 후 Counter 사용

from collections import Counter

ans = Counter( sum(done, [])  )

ans.pop(0, None) #del ans[0]

 

 

ans = sorted(list(ans.values()))

for i in ans :

  print(i)

 

2. 주의사항

 

정답을 출력할떄

 

ans.pop[0]은 keyerror 가 발생하고

del ans[0]은 발생하지 않는다

 

ans.pop(0, None) 는 오류가 발생하지 않는 것으로 보아

 

0 이라는 key 가 존재하지 않을때 keyerror가 발생하는 모양인데

 

그렇다면 왜 del ans[0]은 맞다고 채점될까??