leetcode:Peeking Iterator 迭代器继承,父类调用子类方法

Given an Iterator class interface with methods: next() and hasNext(), design and implement a PeekingIterator that support the peek() operation -- it essentially peek() at the element that will be returned by the next call to next().


Here is an example. Assume that the iterator is initialized to the beginning of the list: [1, 2, 3].

Call next() gets you 1, the first element in the list.

Now you call peek() and it returns 2, the next element. Calling next() after that still return 2.

You call next() the final time and it returns 3, the last element. Calling hasNext() after that should return false

 

题目大意是给你一个迭代器的接口,接口支持两个方法,next(), hasNext(),next()返回下一个元素,指针后移一位,hasNext返回是否还有下一个元素,要求在这个接口上做一层封装,使得支持peek()方法,即获取下一个元素但是不移动指针。

 

题目本身并没有什么难度,增加一个cache缓存一下next()返回值作为peek()的返回值,并标记这个元素已经在缓存里就可以了。

题目有个容易出错的地方是新的peekingIterator需要从iterator类继承,然而父类已经有next()方法了,所以在子类直接用next()得到的是子类的next(),要调用题目给的接口需要加上域操作符iterator::next(),另外编译类时,从上往下编译声明,之后再编译函数体,这个顺序在这里也得以体现。事实证明这些东西没写过还是很容易出错的- -!

// Below is the interface for Iterator, which is already defined for you.
// **DO NOT** modify the interface for Iterator.
class Iterator {
    struct Data;
    Data* data;
public:
    Iterator(const vector<int>& nums);
    Iterator(const Iterator& iter);
    virtual ~Iterator();
    // Returns the next element in the iteration.
    int next();
    // Returns true if the iteration has more elements.
    bool hasNext() const;
};


class PeekingIterator : public Iterator {
public:
    PeekingIterator(const vector<int>& nums) : Iterator(nums) {
        // Initialize any member here.
        // **DO NOT** save a copy of nums and manipulate it directly.
        // You should only use the Iterator interface methods.
        
    }

    // Returns the next element in the iteration without advancing the iterator.
    int peek() {
        if(flag_has_next){
            return cache;
        }else{
            flag_has_next=true;
            cache=Iterator::next();
            return cache;

        }
    }

    // hasNext() and next() should behave the same as in the Iterator interface.
    // Override them if needed.
    int next() {
         if(flag_has_next){
            flag_has_next=false;
            return cache;
        }else{
            return Iterator::next();                                             
    
    }
    }

    bool hasNext() const {
        if(flag_has_next)
            return true;
        else
            return Iterator::hasNext();                                       
        }
    
private:
    bool flag_has_next=false;
    int cache;
};

错点:当执行完peek()之后,不能直接调用子类的hasNext()方法,因为这是可能子类迭代器已经到达末尾了,即,子类hasnext()为false,但是父类还没有到达末尾。所以要先判断一下。



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值