section3.9 봉우리

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

 

 

import sys
sys.stdin = open("input.txt", "rt")
n = int(input())
mlist = [list(map(int, input().split())) for _ in range(n)]
cnt = 0

for i in range(n):
    mlist[i].insert(0, 0)
    mlist[i].insert(n+1, 0)

addzero = [0 for i in range(n+2)]
mlist.insert(0, addzero)
mlist.insert(n+1, addzero)

for x in range(1, n+1):
    for y in range(1, n+1):
        if mlist[x][y] > mlist[x-1][y] and mlist[x][y] > mlist[x][y-1] and mlist[x][y] > mlist[x][y+1] and mlist[x][y] > mlist[x+1][y]:
            cnt += 1

print(cnt)

 

처음에는 0, n-1 행/열만 따로 추출해서 숫자 비교하려고 했는데 너무 복잡해졌다

차라리 0을 추가해 어느 쪽으로 비교해도 가능하게 코드를 짰다

 

정답

 

import sys
sys.stdin = open("input.txt", "rt")
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)

 

 

dx, dy를 설정해두고 코드를 작성하니 내가 짰던 것처럼 길어질 필요가 없었다

자주 봐서 익숙해지기

 

 

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

0으로 행을 추가하는 방법도 간단했다

이것도 익숙해지도록 자주 보기