70-tgyuuAn

[tgyuuAn]

🔗 문제 링크

탑 보기

✔️ 소요된 시간

1시간 20분

✨ 수도 코드

문제 보자마자 한달 전에 풀었던 옥상 정원 꾸미기 문제가 생각나서 쉽게 풀 수 있었습니다.

저 문제 안풀어봤었으면 쉽게 못풀었을 것 같아요.

일단 문제 제한에서

1 <  N, L < 10000

으로 주어졌기 떄문에 최대 시간 복잡도는 &N(log(n))& 을 넘기면 안됩니다!

그렇기 때문에 처음에는 힙을 떠올렸었어요.

힙으로 가장 높이가 작은 것부터 하나씩 차곡차곡 뽑아서 갱신하는 굉장한 아이디어를 가졌었는데,

아래와 같이 볼 수 있는 최대 개수도 같이 반환해야하더라고요.

코드까지 20분 정도 깔짝거리다가 시간만 날렸습니다..

그래서 그럼 &N(log(n))& 이내로 이 문제를 풀 수 있는 해법은 뭐가 있을까? 하고 생각하다가 저 문제가 떠올랐어요.

스택을 사용해서 최대 &O(n)& 으로 풀 수 있답니다.

이 문제를 푸는 데 핵심 키는 다음과 같습니다.

스택을 이용하는데, 스택에 담을 때에는 반드시 내림차순으로 담는 것 입니다.

문제에서 주어진 예시를 들어보겠씁니다.

8
3 7 1 6 3 5 1 7

가장 먼저 보이는 건물은 3입니다.

이 때 스택 stack = [] 은 비어있으므로 그냥 담아줍니다.

그 다음으로 보이는 건물은 7이에요.

이 때 스택을 확인해줍니다. stack = [3]

스택에 가장 마지막 항이 3이네요.

근데 우리는 아까 스택을 반드시 내림차순으로 담는다 라고 했어요.

7이 3보다 크니까 스택에서 3을 제거해줍니다. (여러 개의 항이 있을 경우 7보다 더 큰수가 나올 떄 까지 마지막 항을 pop 해요)

그리고 다시 7을 넣어줍니다. stack = [7]

그 다음으로 보이는 건물은 1 입니다.

다행히 1은 7보다 작네요. 스택에 그냥 넣을 수 있겠어요.

stack = [7, 1]

이 때 스택에 들어갔으므로 3번째 건물은 2번째 건물을 볼 수 있다는 뜻과도 같아요.

볼 수 있는 건물들 리스트를 따로 만들어서 3번째 건물이 2번째를 볼 수 있다고 저장해주겠습니다.

can_see = [0, 0, 2 ... ]

그 다음은 6이 나왔네요.

근데 현재 스택은 stack = [7, 1] 상태라서 1은 6보다 작아요.

그럼 pop 해야겠죠?

그래서 stack = [7, 6] 상태가 되었어요.

그럼 뭐다? 4번째 건물은 2번째 건물을 볼 수 있다 ..!

can_see = [0, 0, 2, 2 ... ]

이런 식으로 쭉 끝까지 가면 됩니다.

근데 여기서 끝내면 안되어요.

우리는 건물을 왼쪽 -> 오른쪽으로만 볼 수 있는 것이 아니고,

왼쪽 <- 오른쪽 으로도 볼 수 있잖아요.

그래서 반대로도 와야해요.

마찬가지로 반대로 쭉쭉 오면서..

스택에 들어갈 때 마다 가장 가까운 볼 수 있는 건물을 갱신해주면 됩니다!

히히힣

이런 유형의 스택 문제가 자주 나오네요.

히스토그램 여기서도 나왔던 것 같은데..

📚 새롭게 알게된 내용

[H0ngJu]

오른쪽 왼쪽 건물 보는게 마치 저번 가희와 탑 문제 같아서 비슷하게 접근했습니다.

그냥 왼쪽에서 바라본 경우와 오른쪽에서 바라본 경우를 각각 계산해주었습니다. 근데 역시 시간복잡도가 N^2이 되어서 시간초과가 나더라고요. 시간 복잡도를 낮출 방법이 떠오르지 않아서 PR 참고하였습니다!

이런류의 문제도 스택으로 풀 수 있군요 .. 흠 스택 문제를 다시 돌아볼 필요성을 느꼈습니다. 옥상정원 꾸미기도 한번 도전해보겠슴다!

시간 초과난 코드

import sys

def input() : return sys.stdin.readline().rstrip()

N = int(input())
buildings = list(map(int, input().split()))

def cnt_buildings(n):
    can_view = []
    building_height = buildings[n]

    # 왼쪽
    for i in range(n-1, -1, -1):
        if buildings[i] > building_height:
            can_view.append(i)
            building_height = buildings[i]
    # 오른쪽
    building_height = buildings[n]
    for i in range(n+1, N):
        if buildings[i] > building_height:
            can_view.append(i)
            building_height = buildings[i]

    tmp = float('inf')
    if not can_view:
        return 0,
    else:
        for b in can_view:
            if abs(b-n) < tmp:
                tmp = abs(b-n)
                min_num_building = b
        return len(can_view), min_num_building+1

for i in range(N):
    print(*cnt_buildings(i))

[alstjr7437]

아니 왼쪽 오른쪽 한번씩 처리를 해줘야 했군요..

처음에는 각 탑들이 왼쪽 오른쪽 확인해서 하면 되지 않나 5초 정도 생각하다가 제한 보고 바로 다른거 생각하는데

도저히 생각이 안나고 그림만 그리다가 시간이 20분 지나서 풀이 참고했네요...

블로그와 같이 바로 확인하고 빼버리면 조금 더 보기 편해진 것 같아요!