[Python] 백준_단지번호붙이기(2667)

문제

 

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

2667번: 단지번호붙이기

 

 

풀이

 

스위프트로 풀었던 문제였지만 파이썬에 조금 익숙해지기 위해서 다시 풀어봄. 쉬운 BFS문제였음.

 

한 칸씩 확인해주면서 단지가 나오면 해당 단지안에 1이 몇개인지 확인하고 총 단지가 몇개인지 확인해서 출력하기만 하면됨.

n = int(input())
board = []
count = 0
result = []
dx = [1,0,-1,0]; dy = [0,1,0,-1]
visited = [[False for _ in range(n)] for _ in range(n)]
for _ in range(n):
    board.append(list(input()))
for i in range(n):
    for j in range(n):
        if board[i][j] == "1" and visited[i][j] == False :
            queue = [(i,j)]
            visited[i][j] = True
            count += 1
            temp = 1
            while queue:
                cur = queue.pop(0)
                for k in range(0,4):
                    newX = cur[0] + dx[k]; newY = cur[1] + dy[k]
                    if newX < 0 or newY < 0 or newX >= n or newY >= n:
                        continue
                    if board[newX][newY] == "0" or visited[newX][newY] == True:
                        continue
                    temp += 1
                    queue.append((newX,newY))
                    visited[newX][newY] = True
            result.append(temp)
print(count)
result.sort()
for i in range(len(result)):
    print(result[i])

 

 

_____________________________________________________________________________________________________

 

 

스위프트로 푸는게 훨씬 쉬움. 코드를 그대로 옮기는 것도 힘듦