[골드 V] 1245. 농장 관리 - 파이썬

1245번: 농장 관리

1245번: 농장 관리

 

내 풀이(52ms)

import sys
input = sys.stdin.readline

def check(i, j):
    global n, m, mount
    # 8가지 방향
    dires = [(-1, 0), (1, 0), (0, -1), (0, 1), (1, 1), (-1, -1), (1, -1), (-1, 1)]
    max_val = maps[i][j]
    visited[i][j] = 1

    # 인접한 격자 탐색
    for dx, dy in dires:
        ni, nj = i + dx, j + dy
        if 0 <= ni < n and 0 <= nj < m:
            # 인접한 격자보다 높이가 낮으면 False
            if max_val < maps[ni][nj]:
                mount = False
            # 같은 높이의 인접한 격자 모두 check
            if max_val == maps[ni][nj] and not visited[ni][nj]:
                check(ni, nj)

n, m = map(int, input().rstrip().split())
maps = [list(map(int, input().rstrip().split())) for _ in range(n)]
visited = [[0] * m for _ in range(n)]
count = 0

for i in range(n):
    for j in range(m):
        if maps[i][j] and not visited[i][j]:
            mount = True
            check(i, j)  
            if mount:
                # True인 경우: 인접한 격자 중 자신보다 높은 것이 없다는 것
                count += 1

print(count)