Python | Leetcode Python题解之第284题窥视迭代器

题目:

Python | Leetcode Python题解之第284题窥视迭代器_第1张图片

题解:

class PeekingIterator:
    def __init__(self, iterator):
        self.iterator = iterator
        self._next = iterator.next()
        self._hasNext = iterator.hasNext()

    def peek(self):
        return self._next

    def next(self):
        ret = self._next
        self._hasNext = self.iterator.hasNext()
        self._next = self.iterator.next() if self._hasNext else 0
        return ret

    def hasNext(self):
        return self._hasNext

你可能感兴趣的:(分享,Python,Leetcode,题解)