[예제] [파이썬] [탐색] 봉우리

지도 정보가 N*N 격자판에 주어집니다. 각 격자에는 그 지역의 높이가 쓰여있습니다. 각 격자
판의 숫자 중 자신의 상하좌우 숫자보다 큰 숫자는 봉우리 지역입니다. 봉우리 지역이 몇 개
있는 지 알아내는 프로그램을 작성하세요.
격자의 가장자리는 0으로 초기화 되었다고 가정한다.
만약 N=5 이고, 격자판의 숫자가 다음과 같다면 봉우리의 개수는 10개입니다.

 

 

▣ 입력설명
첫 줄에 자연수 N이 주어진다.(1<=N<=50)
두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다. 각 자연수는 100을 넘지 않는
다.

▣ 출력설명
봉우리의 개수를 출력하세요.

▣ 입력예제 1
5
5 3 7 2 3
3 7 1 6 1
7 2 5 3 4
4 3 6 4 1
8 7 3 5 2

▣ 출력예제 1
10

 

---

 

import sys

sys.stdin=open('input.txt','r')

dx=[-1,0,1,0]
dy=[0,1,0,-1]


if __name__=="__main__":
    n=int(input())
    arrays=[list(map(int, input().split())) for _ in range(n)]
    arrays.insert(0,[0]*n)
    arrays.append([0]*n)
    for arr in arrays:
        arr.insert(0,0)
        arr.append(0)
    # ----------------- 여기까지 준비 완료 ----------------------------
        
    cnt=0
    for i in range(1,n+1):
        for j in range(1,n+1):
            if all(arrays[i][j]>arrays[i+dx[k]][j+dy[k]] for k in range(4)):    # 상하좌우를 다 살핀다.
                cnt+=1
                
    print(cnt)