SWEA: 1206 View

문제

SW Expert Academy

 

문제 풀이

 빌딩의 높낮이가 주어질 때, 하나의 빌딩을 기준으로 좌측으로 2칸, 우측으로 2칸의 빌딩 중 낮은 빌딩이 있는 경우를 찾는 문제이다. 이때 빌딩의 높이가 좌우측의 빌딩 중 가장 높은 빌딩보다 높은 층은 전망권이 확보된 세대수이다. 주어진 입력값에서 전망권이 확보된 세대수를 반환하면 된다.

 

코드

for test_case in range(1, 11):
    length = int(input())
    buildings = list(map(int, input().split()))
    answer = 0

    for idx in range(2, length - 2):
        cur_height = buildings[idx]
        left = max(buildings[idx - 2:idx])
        right = max(buildings[idx + 1: idx + 3])

        if left > cur_height or right > cur_height:
            continue

        answer += cur_height - max(left, right)

    print('#' + str(test_case), answer)

 좌우측으로 빈칸인 0이 주어지므로, `(2, length - 2)`의 인덱스를 확인하면 된다.