[알고리즘] 봉우리

(문제) 봉우리


지도 정보가 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
7
27 33 80 73 19 23 15
48 84 61 3 36 9 62
87 57 3 14 69 17 22
17 57 86 21 85 51 82
7 94 66 21 19 41 82
27 5 59 28 26 77 64
5 46 4 63 76 99 8
11

해결방법

  1. 상하좌우를 나타낼 수 있는 행, 열을 나타내는 리스트를 생성하여 상대적 값을 삽입한다.
  2. 반복문을 통해 접근한 값의 상하좌우가 접근한 값보다 큰 것이 없을 경우 count를 한다.

코드

n = int(input())

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

for i in range(len(arr)):
    # 앞 뒤 0 으로 채우기
    arr[i].append(0)
    arr[i].insert(0, 0)

# 앞 뒤 0 리스트로 초기화
arr.append([0]*(n+2))
arr.insert(0, [0]*(n+2))


# 상하좌우
# 행
dx = [-1, 1, 0, 0]
# 열
dy = [0, 0, -1, 1]

count = 0

for i in range(1, n+1):
    for j in range(1, n+1):
        for k in range(4):
            # 다른 것보다 작을 경우
            if arr[i][j] <= arr[i+dx[k]][j+dy[k]]:
                break
        else:
            count = count + 1

print(count)

댓글남기기