문제 / 봉우리 / Python 파이썬

 

문제 제목: 봉우리

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

---

 

그림대로 구현하기

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

for i in range(n):
    arr[i].insert(0,0)
    arr[i].append(0)

arr.insert(0,[0]*(n+2))
arr.append([0]*(n+2))

for i in arr:
    print(i)

 

나의 풀이

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

for i in range(n):
    arr[i].insert(0,0)
    arr[i].append(0)

arr.insert(0,[0]*(n+2))
arr.append([0]*(n+2))
cnt = 0
for i in range(1,n+1):
    for j in range(1,n+1):
        current = arr[i][j]
        up = arr[i-1][j]
        down = arr[i+1][j]
        left = arr[i][j-1]
        right = arr[i][j+1]

        if current > up and current > down and current > right and current > left:
            cnt += 1
print(cnt)

모범답안

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)

---