티스토리 뷰

728x90

문제

www.acmicpc.net/problem/1012

 

1012번: 유기농 배추

차세대 영농인 한나는 강원도 고랭지에서 유기농 배추를 재배하기로 하였다. 농약을 쓰지 않고 배추를 재배하려면 배추를 해충으로부터 보호하는 것이 중요하기 때문에, 한나는 해충 방지에 

www.acmicpc.net

풀이

dfs로 풀어줬다.

sys.setrecursionlimit(100000) 안해주면 재귀오류? 뜸

 

소스코드

from collections import deque
import sys
sys.setrecursionlimit(100000)
input = sys.stdin.readline

def dfs(y, x):
    dx = [-1, 1, 0, 0]
    dy = [0, 0, -1, 1]
    arr[y][x] = 0
    for i in range(4):
        nowx, nowy = dx[i]+x, dy[i]+y
        if 0<=nowx<m and 0<=nowy<n and arr[nowy][nowx]:
            dfs(nowy, nowx)

    
t = int(input())
for _ in range(t):
    m, n, k = map(int, input().split())
    arr = [[0]*(m) for _ in range(n)]
    for i in range(k):
        a, b = map(int, input().split())
        arr[b][a]=1
    cnt = 0
    for i in range(n):
        for j in range(m):
            if arr[i][j]==1:
                cnt += 1
                dfs(i, j)
    print(cnt)
반응형