[SW Expert Academy, SWEA 1206] [S/W 문제해결 기본] 1일차 - View (python)

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AV134DPqAA8CFAYh&categoryId=AV134DPqAA8CFAYh&categoryType=CODE&problemTitle=&orderBy=RECOMMEND_COUNT&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=1

SW Expert Academy

---

※ SW Expert 아카데미의 문제를 무단 복제하는 것을 금지합니다.

728x90

---

내 생각

해당 건물 왼쪽에 인접한 2개의 건물과 오른쪽에 인접한 2개의 건물 중 가장 높은 건물보다 해당 건물이 더 높으면, 높은 만큼 조망권이 확보된다.

for test_case in range(1, 11):
    # 건물의 개수 n
    n = int(input())

    result = 0
    arr = list(map(int, input().split()))
    for i in range(2, n -2):
        sub = arr[i] - max(arr[i - 2], arr[i - 1], arr[i + 1], arr[i + 2])
        if sub > 0:
            result += sub

    print("#{} {}".format(test_case, result))