🏄‍♀️ 코딩테스트/🐍 Python

[ 백준 1743 해설 ] ( python ) 음식물 피하기

Po_tta_tt0 2022. 7. 18. 15:51
반응형

 

📚 음식물 피하기

 

문제

코레스코 콘도미니엄 8층은 학생들이 3끼의 식사를 해결하는 공간이다. 그러나 몇몇 비양심적인 학생들의 만행으로 음식물이 통로 중간 중간에 떨어져 있다. 이러한 음식물들은 근처에 있는 것끼리 뭉치게 돼서 큰 음식물 쓰레기가 된다. 

이 문제를 출제한 선생님은 개인적으로 이러한 음식물을 실내화에 묻히는 것을 정말 진정으로 싫어한다. 참고로 우리가 구해야 할 답은 이 문제를 낸 조교를 맞추는 것이 아니다. 

통로에 떨어진 음식물을 피해가기란 쉬운 일이 아니다. 따라서 선생님은 떨어진 음식물 중에 제일 큰 음식물만은 피해 가려고 한다. 

선생님을 도와 제일 큰 음식물의 크기를 구해서 “10ra"를 외치지 않게 도와주자.

 

입력

첫째 줄에 통로의 세로 길이 N(1 ≤ N ≤ 100)과 가로 길이 M(1 ≤ M ≤ 100) 그리고 음식물 쓰레기의 개수 K(1 ≤ K ≤ N×M)이 주어진다.  그리고 다음 K개의 줄에 음식물이 떨어진 좌표 (r, c)가 주어진다.

좌표 (r, c)의 r은 위에서부터, c는 왼쪽에서부터가 기준이다. 입력으로 주어지는 좌표는 중복되지 않는다.

 

출력

첫째 줄에 음식물 중 가장 큰 음식물의 크기를 출력하라.

 

예제 입력 1 복사

3 4 5
3 2
2 2
3 1
2 3
1 1

예제 출력 1 복사

4

힌트

# . . .
. # # .
# # . .

위와 같이 음식물이 떨어져있고 제일큰 음식물의 크기는 4가 된다. (인접한 것은 붙어서 크게 된다고 나와 있음. 대각선으로는 음식물 끼리 붙을수 없고 상하좌우로만 붙을수 있다.)

 

 

 

 

 

✍ 접근

  • BFS

 

 

 

 

 

 

 

정답코드

# ✨ 입력
import sys
from collections import deque
input = sys.stdin.readline
N,M,K = map(int,input().split())
board = [[0]*M for _ in range(N)]
res = 0
for _ in range(K):
    a,b = map(int,input().split())
    board[a-1][b-1] = 1
    
dx = [-1,0,1,0]
dy = [0,-1,0,1]

# ✨ BFS
def BFS(x,y):
    dq = deque([])
    dq.append((x,y))
    board[x][y] = -1
    cnt = 1
    while dq:
        xx,yy = dq.popleft()
        for i in range(4):
            nx = xx + dx[i]
            ny = yy + dy[i]
            if 0<=nx<N and 0<=ny<M and board[nx][ny] == 1:
                board[nx][ny] = -1
                cnt+=1
                dq.append((nx,ny))
    return cnt
        
# ✨ 탐색
for i in range(N):
    for j in range(M):
        if board[i][j] == 1:
            res = max(res, BFS(i,j))
print(res)

해설

입력

  • 음식물쓰레기가 있는 곳은 1로 아닌곳은 그냥 0으로 board를 만든다

탐색

  • board를 탐색하면서 음식물 쓰레기가 발견되면 BFS

BFS

  • 4방위를 돌면서 음식물 쓰레기가 근처에 있으면(큰 덩어리라면) cnt에 1씩 더해주고 방문을 기록하기 위해 board를 -1로 업데이트해준다.
  • while문이 끝나면 cnt를 return하고 res를 구한다

 

 

 

⭐ 배움

  • 익숙한 BFS 문제.
 

 

반응형