[LeetCode]next greater element 1 (완료) - 고민해볼 것

[yeonghwanjeon]

• 문제 링크

• 문제 풀이

  • Stack 활용 문제
  • 왼쪽부터 보면서, 오른쪽 값들 중 제일 큰 값 저장
  • Dictionary 형태로 저장해서 마지막에 return.

  • O(n)으로 가능할 듯

    class Solution(object):
    def nextGreaterElement(self, findNums, nums):
    """
    :type findNums: List[int]
    :type nums: List[int]
    :rtype: List[int]
    """
    nexts = {}
    stack = []
    for n in nums :
        while stack and n > stack[-1] :
            nexts[stack.pop()] = n
        stack.append(n)
    
    while stack :
        nexts[stack.pop()] = -1
    
    return [nexts[n] for n in findNums]