019 - 봉우리

지도 정보가 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('in5.txt', 'rt')

n = int(input())
a = [list(map(int, input().split())) for _ in range(n)]
# 4방향 찾기에서 for 문으로 코딩량을 줄일 수 있다.
dx = [1, 0, -1, 0]
dy = [0, 1, 0, -1]
cnt = 0

for i in range(n):
    for j in range(n):
        for k in range(4):
            x, y = i+dx[k], j+dy[k]
            if 0 <= x < n and 0 <= y < n and a[i][j] <= a[x][y]:
                break
        else:
            cnt += 1
print(cnt)

 

 

문제에서 설명한 '0' 으로 둘러싸인 2중 배열을 처러하는 코딩 예시.

 

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
        # all 함수가 아래 코딩과 동일한 결과, all 함수는 모든 조건이 만족해야 실행된다.
        # for k in range(4):
        #     if a[i][j] <= a[i+dx[k]][j+dy[k]]:
        #         break
        # else:
        #     cnt += 1
print(cnt)