[백준 14716번] 파이썬 - 현수막

백준 14716 - 현수막

시간 제한 2초, 메모리 제한 512MB

# 조건

• 저번 학기 영상처리 수업을 듣고 배웠던 지식을 최대한 응용 해보고 싶은 혁진이는 이 현수막에서 글자가 몇 개인지 알아보는 프로그램을 만들려 한다.
• 혁진이는 우선 현수막에서 글자인 부분은 1, 글자가 아닌 부분은 0으로 바꾸는 필터를 적용하여 값을 만드는데 성공했다.
• 그런데 혁진이는 이 값을 바탕으로 글자인 부분 1이 상, 하, 좌, 우, 대각선으로 인접하여 서로 연결되어 있다면 한 개의 글자라고 생각만 하였다.
• 혁진이가 필터를 적용하여 만든 값이 입력으로 주어졌을 때, 혁진이의 생각대로 프로그램을 구현하면 글자의 개수가 몇 개인지 출력하여라.

입력

• 첫 번째 줄에는 현수막의 크기인 M와 N가 주어진다. (1 ≤ M, N ≤ 250)
• 두 번째 줄부터 M+1 번째 줄까지 현수막의 정보가 1과 0으로 주어지며, 1과 0을 제외한 입력은 주어지지 않는다.

출력

• 혁진이의 생각대로 프로그램을 구현했을 때, 현수막에서 글자의 개수가 몇 개인지 출력하여라.

# 접근 방법

• BFS로 풀어주면 된다.
• 몇 개의 그래프가 있는지라고 생각하면 더 쉽게 풀 수 있다.
• 이 때, 상하좌우가 아닌 대각선까지 delta로 탐색해주면 된다.
• 0, 0에서 시작하여 1을 발견한 경우 BFS를 수행하여 하나의 집단으로 만들고 VISITED를 체크해준며 해당 탐색이 끝난 경우 RESULT += 1을 해준다.
• 풀이시간 : 10분

import sys
sys.stdin = open('input.txt')
input = sys.stdin.readline

from collections import deque

def bfs(bi, bj):
    q = deque()
    q.append((bi, bj))
    while q:
        si, sj = q.popleft()
        for d in range(8):
            ni, nj = si+di[d], sj+dj[d]
            can = check_range(ni, nj)
            if can:
                visited[ni][nj] = True
                q.append((ni, nj))

def check_range(ci, cj):
    if 0<=ci<N and 0<=cj<M and arr[ci][cj] == 1 and visited[ci][cj] == False:
        return True
    return False

N, M = map(int, input().split())
arr = [list(map(int, input().split())) for _ in range(N)]
visited = [[False] * M for _ in range(N)]
result = 0
# 상 하 좌 우 좌상 우상 좌하 우하
di, dj = [-1, 1, 0, 0, -1, -1, 1, 1], [0, 0, -1, 1, -1, 1, -1, 1]

for i in range(N):
    for j in range(M):
        if arr[i][j] == 1:
            if visited[i][j] == False:
                visited[i][j] = True
                bfs(i, j)
                result += 1

print(result)