[알고리즘]봉우리

문제

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

입력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

[내 코드]

def find_max(num,N):
    N=N+2
    is_peak=[[0 for _ in range(N)] for _ in range(N)]
    count=0
    for i in range(1,N-1):
        for j in range(1,N-1):
            if num[i][j]>num[i-1][j] and num[i][j]>num[i+1][j] and num[i][j]>num[i][j-1] and num[i][j]>num[i][j+1]  and is_peak[i][j]==0:
                is_peak[i][j]=1
            #리스트 중 맥스를 찾으면 주변값이 같아도 인정되므로
            #해당값 num[i][j]의 값이 주변보다 커야함.(같은 것은 인정 ㄴㄴ)
    for i in range(N):
        for j in range(N):
            if is_peak[i][j]==1:
                count+=1
    return count
    


N=int(input())
numList=[[0 for _ in range(N+2)] for _ in range(N+2)]

a=[list(map(int,input().split())) for _ in range(N)]
for i in range(N):
    for j in range(N):
        numList[i+1][j+1]=a[i][j]

num=find_max(numList,N)
print(num)

[해설코드]

n=int(input())
dx=[-1,0,1,0]
dy[0,1,0,-1]
#dx는 행방향, dy는 열방향-> ex) (-1,0) 9시 방향
a=[list(map(int,input().split())) for _ in range(n)]
a.insert(0,[0]*n)
a.append([0]*n)
for x in a:
	x.insert(0,0)
    x.append(0)
    
cnt=0
for i in range(1,n+1):
	for j in range(1,n+1):
    #상하좌우탐색
    	if all(a[i][j]>a[i+dx[k]][j+dy[k]] for k in range(4)):
        	cnt+=1
print(cnt)