[SWEA] 1206. View (S/W 문제해결 기본 - 1일차) / Python 코드

학기가 끝나가면서 다음 분기 준비로 코딩 테스트가 급박해지는 시점, 시간을 내어서 해보자.

곧 코테 언어도 JAVA로 바꿔서 연습할 듯 한데

이번 주는 급한 마음에 파이썬으로!

 

SW Expert Academy의 D3 난이도 문제로 전에 풀었던 D2 문제보다도 쉬웠다.

 

    T = 10
    for test_case in range(1, T + 1):
        N = int(input())
        height = list(map(int, input().split()))
        result = 0

        for j in range(2, N - 2):
            cur = height[j - 2:j + 3]
            cur.sort(reverse=True)
            if cur[0] == height[j]:
                result += cur[0] - cur[1]

        print(f"#{test_case} {result}")

 

제약 사항 확인과 input값만 잘 확인하면 쉽게 풀 수 있을 것이다.