SWEA - View(1206)

SW Expert Academy

 

문제

강변에 빌딩들이 옆으로 빽빽하게 밀집한 지역이 있다.

이곳에서는 빌딩들이 너무 좌우로 밀집하여, 강에 대한 조망은 모든 세대에서 좋지만 왼쪽 또는 오른쪽 창문을 열었을 때 바로 앞에 옆 건물이 보이는 경우가 허다하였다.

그래서 이 지역에서는 왼쪽과 오른쪽으로 창문을 열었을 때, 양쪽 모두 거리 2 이상의 공간이 확보될 때 조망권이 확보된다고 말한다.

빌딩들에 대한 정보가 주어질 때, 조망권이 확보된 세대의 수를 반환하는 프로그램을 작성하시오.
 
아래와 같이 강변에 8채의 빌딩이 있을 때, 연두색으로 색칠된 여섯 세대에서는 좌우로 2칸 이상의 공백이 존재하므로 조망권이 확보된다. 따라서 답은 6이 된다.

A와 B로 표시된 세대의 경우는 왼쪽 조망은 2칸 이상 확보가 되었지만 오른쪽 조망은 한 칸 밖에 확보가 되지 않으므로 조망권을 확보하지 못하였다.

C의 경우는 반대로 오른쪽 조망은 2칸이 확보가 되었지만 왼쪽 조망이 한 칸 밖에 확보되지 않았다.
 
[제약 사항]

가로 길이는 항상 1000이하로 주어진다.

맨 왼쪽 두 칸과 맨 오른쪽 두 칸에는 건물이 지어지지 않는다. (예시에서 빨간색 땅 부분)

각 빌딩의 높이는 최대 255이다.

 

 

접근방식

해당하는 건물 왼쪽 2칸, 오른쪽 2칸까지 중에, 이 건물의 높이가 제일 높으면 조망권 확보

ans = 0으로 설정

맨 왼쪽 두 칸과 오른쪽 두  칸은 건물이 지어지지 않으므로 높이가 0 -> 배열은 (2부터 N-2)까지 돌림

buildings[i]의 주변 2칸(buildings[i-2], buildings[i-1], buildings[i+1], buildings[i+2])중 max값을 around_max 변수에 할당

ex) buildings[i] =5, buildings[i-2] = 3, buildings[i-1] = 2, buildings[i+1] = 2, buildings[i+2] = 1이면 around_max = 3

높이 차이를 알기 위해 tmp라는 변수에 buildings[i] - around_max를 할당(tmp = 5 - 3 = 2)

tmp값이 0보다 크면(주변으로 2칸의 건물들보다 해당 건물높이가 높음) 해당 tmp값을 ans에 더함

for문을 다 돌면 조망권 확보한 개수가 나옴

 

코드

T = int(input())
for t in range(1, T + 1):
    N = int(input())
    buildings = list(map(int, input().split()))
    ans = 0
    for i in range(2, N-2):
        around_max = max(buildings[i-2], buildings[i-1], buildings[i+1], buildings[i+2])
        tmp = buildings[i] - around_max
        if tmp > 0:
            ans += tmp
    print('#'+str(t), ans)