[SWEA/D3] 1206 [S/W 문제해결 기본] 1일차 - View

문제 바로가기

SW Expert Academy

나의 코드 및 설명 01

• 현재 건물을 기준으로 양 옆의 네 건물들을 확인하여 조망권이 확보되었는지 파악하는 코드를 작성하였다.

for test_case in range(1, 11):
    ans = 0
    n = int(input()) # 건물의 개수
    apt = list(map(int, input().split()))
    for i in range(2, len(apt)):
    	#첫 번째 건물일 경우 뒤에 있는 건물 두개만 확인
        if i == 2:
        	#현재 건물이 다음 건물과 그 다음 건물보다 클 경우에만 해당됨
            if apt[i] > apt[i+1] and apt[i] > apt[i+2]:
            	#현재 건물의 높이에서 다음 건물과 다다음 건물 중 큰 것을 빼면 조망권이 확보된 세대수이다.
                ans += apt[i]-max(apt[i+1],apt[i+2])
            else: 
                continue
        elif i >= len(apt)-2:
            if apt[i] > apt[i-1] and apt[i] > apt[i-2]:
                ans += apt[i]-max(apt[i-1],apt[i-2])
            else:
                continue
        else:
            if (apt[i] > apt[i-1] and apt[i] > apt[i-2]) and (apt[i] > apt[i+1] and apt[i] > apt[i+2]):
                ans += apt[i] - max(apt[i-1],apt[i-2],apt[i+1],apt[i+2])
            else:
                continue
    print("#{} {}".format(test_case, ans))