Today's Challenge

 

class NestedIterator:
    def __init__(self, nestedList: [NestedInteger]):
        self.it = iter(nestedList)
    
    def next(self) -> int:
        val = next(self.it)
        
        if val.isInteger():
            return val.getInteger()
    
        self.it = chain(va.getList(), self.it)
        
        return self.next()
    
    def hasNext(self) -> bool:
        val: Optional[NestedInteger] = next(self.it, None)
        
        if val is None:
            return False
        elif val.isInteger():
            self.it = chain([val], self.it)
            return True
        else:
            self.it = chain(val.getList(), self.it)
            return self.hasNext()

https://leetcode.com/problems/flatten-nested-list-iterator/

Flatten Nested List Iterator - LeetCode

오늘은 파이썬의 새로운 기법들을 많이 사용해서 푸는 문제였다.
iter라는 걸 이렇게 활용할 수 있는걸 처음 알게 됨.