봉우리 - python

<문제>

지도 정보가 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

 

<코드>

n = int(input())
L=[list(map(int,input().split())) for _ in range(n)]

cnt = 0
check1 = 0
check2 = 0
for i in range(n):
    for j in range(n):
        a = L[i][j]
        if 0<j<n-1:
            lp = L[i][j-1]
            rp = L[i][j+1]
        elif j == 0:
            lp = 0
            rp = L[i][j+1]
        elif j == n-1:
            lp = L[i][j-1]
            rp = 0
        if a>lp and a>rp:
            check1 = 1
        if 0<i<n-1:   
            up = L[i-1][j]
            dp = L[i+1][j]
        elif i == 0:
            up = 0
            dp = L[i+1][j]
        elif i == n-1:
            up = L[i-1][j]
            dp = 0
        if a>up and a>dp:
            check2 = 1
        if check1==check2==1:
            cnt+=1
        check1=0
        check2=0
print(cnt)