리트코드 739번 일일온도 내풀이와 상길선생님 풀이 비교

• 문제 요구사항은 사이트에 들어가서 확인하자.

내풀이

from collections import deque
class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        temperatures = deque(temperatures)
        answer = []
        for i in range(len(temperatures)):
            a = temperatures.popleft()
            while True:
                j = 0 
                if temperatures[j] <= a:
                    j += 1
                elif temperatures[j] > a:
                    j += 1
                    answer.append(j)
                    break 
                if j >= len(temperatures):
                    answer.append(0)
                    break

• 가장 처음 고민한 건 for문을 한번만 쓰고 구현하는 법에 대해서 고민했다.
  처음에는 while문만 사용해서 풀려고 했으나 하나씩 다 비교를 해야 하므로 for문에 while문을 넣는 걸로 타협을 봤다.

• 첫번째 반복문에서 앞의 요소를 빼고 다음 while 반복문을 순회하면서 인덱스를 저장하는 변수 j를 선언해서 각 조건마다 j값을 1씩 올리다가 조건에 해당하면 빈 배열에 j를 집어 넣는다.

• 그치만 시간초과가 나서 체점조차 되지 않았다....

상길 선생님의 풀이

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:                    
        answer = [0] * len(T)
        stack = []
        for i, cur in enumerate(T):
            # 현재 온도가 스택 값보다 높다면 정답 처리 
            while stack and cur > T[stack[-1]]:
                last = stack.pop()
                answer[last] = i - last
            stack.append(i)
        return answer 

• 오늘도 역시나 상길 선생님의 풀이를 보면 현타가 온다. 어떻게 이렇게 풀 수 있을까 나는 독방에 가둬두고 10년동안 이 문제 풀라고 해도 이런 풀이 방법은 못 떠올릴텐데...하는 마음에 그만두고 싶은 마음이 굴뚝 같지만 이렇게 좋은 풀이방법을 익혔으니 개이득이다 라고 생각한다.

• for문 한번에 while문을 두는 것 까지는 내생각과 비슷했다.
  선생님은 enumerate함수를 사용해서 인덱스 값을 저장해서 스택배열에 인덱스 값을 저장하다가 현재 값이 이전 값 보다 큰 값이 나오면 스택에 저장한 인덱스 배열을 빼버리고 비교값의 인덱스 차이값으로 업데이트를 해버린다.

• 항상 나의 풀이 방식은 브루트 포스트 인데 이렇게 다양한 방법으로 생각하는 훈련을 더 해야겠다.