[알고리즘] 봉우리

📌

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

 

📌 input

 

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: 주어진 matrix 가장자리를 어떻게 0으로 초기화 할 것인가?

2: 상하좌우 숫자와 자신을 어떻게 비교할 것인가?

 

 

✔️ 내가 짠 코드 

 

import sys
sys.stdin=open("input.txt", "r")

n=int(input())
new_N=[[0]*(n+2) for _ in range(n+2)] #모든 값이 0으로 초기화된 matrix를 정의한다.
N=[list(map(int, input().split())) for _ in range(n)]
res=0

for i in range(1, n+1):
    for j in range(1, n+1):
        new_N[i][j]=N[i-1][j-1] #새로운 matrix에 주어진 격자판 값을 삽입한다.

for i in range(1, n+1):
    for j in range(1, n+1):
    	#상하좌우 index를 이용하여 비교한다.
        if new_N[i][j]>new_N[i-1][j] and new_N[i][j]>new_N[i][j-1] and new_N[i][j]>new_N[i][j+1] and new_N[i][j]>new_N[i+1][j]:
            res+=1

print(res)

 

 

✔️ 실제 코드

 

#좌우상하
#(i, j) 일때 dx는 i에 dy는 j에 더해준다.
dx=[-1, 0, 1, 0]
dy=[0, 1, 0, -1]
#가장 윗줄에 n개의 0을 추가한다.
N.insert(0, [0]*n)
#가장 아랫줄에 n개의 0을 추가한다.
N.append([0]*n)

#matrix를 돌면서 각 행의 가장 앞, 뒤에 0을 추가한다.
for x in N:
    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)