■코테 중요개념/BFS
[백준 1012] 유기농 배추
영민 박
2020. 5. 28. 23:39
문제 : https://www.acmicpc.net/problem/1012
1012번: 유기농 배추
차세대 영농인 한나는 강원도 고랭지에서 유기농 배추를 재배하기로 하였다. 농약을 쓰지 않고 배추를 재배하려면 배추를 해충으로부터 보호하는 것이 중요하기 때문에, 한나는 해충 방지에 �
www.acmicpc.net
▶ 코드 ( 재귀 ver. )
import sys
sys.setrecursionlimit(10**8) # 10^8 까지 늘림.
T = int(input())
for _ in range(T):
m, n, k = map(int, input().split())
myMap = [[0]*m for _ in range(n)]
for _ in range(k):
a, b = map(int, input().split())
myMap[b][a] = 1
#print(myMap)
visited = [[0]*m for _ in range(n)]
dx = [1,-1,0,0]
dy = [0,0,1,-1]
def isInGraph(r, l): # r: 세로, l: 가로
if r < 0 or r >= n or l < 0 or l >= m:
return False
return True
def bfs(r, l):
visited[r][l] = 1
tx = 0
ty = 0
for i in range(4):
tx = r + dx[i]
ty = l + dy[i]
if isInGraph(tx, ty):
if visited[tx][ty] == 0 and myMap[tx][ty] == myMap[r][l]:
bfs(tx, ty)
ans = 0
for i in range(n): # 세로길이
for j in range(m): # 가로길이
if visited[i][j] == 0 and myMap[i][j] != 0:
bfs(i, j)
ans += 1
print(ans)
# pypy로 제출하면 메모리초과. python3로 제출해야됨
※ python3로 채점해야 메모리초과 X