Today's Challenge

https://leetcode.com/problems/peeking-iterator

내가 문제 이해를 잘 못했나..
오늘은 좀 난해했음 ㅋㅋㅋㅋ 그렇다고 어려운 건 아니었지만!

class PeekingIterator:
    def __init__(self, iterator):
        self.iterator = iterator        
        self.top = self.getTop()
        

    def peek(self):
        return self.top        

    def next(self):
        if self.top:
            top = self.top
            self.top = self.getTop()
            return top     

    def hasNext(self):
        return True if self.top else False
    
    def getTop(self):
        return self.iterator.next() if self.iterator.hasNext() else None