본문 바로가기

파이썬 알고리즘 풀이

19) 봉우리

문제

봉우리 지역이 몇 개 있는 지 알아내는 프로그램을 작성하기

 

조건

지도 정보가 N*N 격자판에 주어집니다.

 

각 격자에는 그 지역의 높이가 쓰여있습니다.

 

각 격자 판의 숫자 중 자신의 상하좌우 숫자보다 큰 숫자는 봉우리 지역입니다.

 

격자의 가장자리는 0으로 초기화 되었다고 가정한다.

 

만약 N=5 이고, 격자판의 숫자가 다음과 같다면 봉우리의 개수는 10개입니다.

제한사항

첫 줄에 자연수 N이 주어진다.(1<=N<=50) 두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다.

 

각 자연수는 100을 넘지 않는다.

 

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

 

문제풀이

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

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

cnt=0

a.insert(0, [0]*N)
a.append([0]*N)

for i in a:
  i.insert(0, 0)
  i.append(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)

'파이썬 알고리즘 풀이' 카테고리의 다른 글

21) 격자판 회문수  (0) 2021.10.22
20) 스도쿠 검사  (0) 2021.10.22
18) 곳감(모래시계)  (0) 2021.10.22
17) 사과나무(다이아몬드)  (0) 2021.10.22
16) 격자판 최대합  (0) 2021.10.22