[파이썬 알고리즘] 봉우리

봉우리

📄 문제

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

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

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

▣ 입력예제
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

▣ 출력예제
10

🧞 풀이

풀이

n = int(input())                                                                # 정수 n 입력 및 할당
a = [list(map(int, input().split())) for _ in range(n)]                         # 값들 입력 받아 배열 할당
a.insert(0, [0] * (n + 2))                                                      # 맨 윗 줄에 0으로 된 행 넣기
a.append([0] * (n + 2))                                                         # 맨 아랫 줄에도 0으로 된 행 넣기
for i in range(1, n+1):                                                         # 1부터 n까지 반복
    a[i].insert(0, 0)                                                           # 각 행 첫 부분에 0 추가
    a[i].append(0)                                                              # 각 행 끝 부분에 0 추가

cnt = 0                                                                         # 봉우리 갯수 세기 위한 cnt 할당

for j in range(1, n+1):                                                         # 1부터 n까지 반복
    for k in range(1, n+1):                                                     # 이중반복문
        p = a[j][k]                                                             # 봉우리 검사를 위한 p 할당
        if p > a[j][k+1] and p > a[j][k-1] and p > a[j+1][k] and p > a[j-1][k]: # p를 상하좌우랑 비교했을 때 모두 큰 값이면
            cnt += 1                                                            # cnt를 1만큼 증가

print(cnt)                                                                      # 계산을 통해 얻은 봉우리 갯수 출력

 

다른 사람 풀이

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

n = int(input())
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)