[Python Algorithm] 탐색&시뮬레이션(2차원 리스트 탐색) - 봉우리

> 탐색 & 시뮬레이션 (2차원 리스트 탐색 편)

 

📖 문제 : 봉우리

지도 정보가 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

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
dx = [0, 1, 0, -1]
dy = [1, 0, -1, 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)

 

⭐️ Point ! ⭐️

• 2차원 리스트에 값 0으로 n개 이루어진 행을 가장 첫 번째에 삽입
  list.insert(0 , [0]*n)
  
  
• 2차원 리스트에 값 0으로 n개 이루어진 행을 가장 마지막에 삽입
  list.append([0]*n)
  
  
• list[i][j] 의 4개의 방향(위, 아래, 왼쪽, 오른쪽) 과 값 비교
  dx = [0, 1, 0, -1]
  dy = [1, 0, -1, 0]     두개의 리스트 활용하기.
  if all(a[i][j]>a[i+dx[k]][j+dy[k]] for k in range(4)):
  
  
• for문을 통해 나오는 모든 값이 참일 경우 참으로 판단하는 all()을 활용한 if문
  => if all(a[i][j]>a[i+dx[k]][j+dy[k]] for k in range(4)):

 

 

 

출처

• 인프런 : 파이썬 알고리즘 문제 풀이