LeetCode "Zigzag Iterator"

zigzag迭代器：实现交替遍历两个向量

最新推荐文章于 2025-02-03 15:37:10 发布

转载最新推荐文章于 2025-02-03 15:37:10 发布 · 74 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/tonix/p/4809038.html

本文介绍了一个zigzag迭代器类，用于交替遍历两个向量，实现复杂的数据处理任务。

Capable to k-vector input too:

class ZigzagIterator {
    int x;    
    int i;
    int max_x;
    vector<vector<int>*> l;
    
    void moveon()
    {
        int oldi= i;
        i = (i + 1) % l.size();
        x += i <= oldi;
    }
public:
    ZigzagIterator(vector<int>& v1, vector<int>& v2) 
    {
        i = x = 0;
        max_x = max(v1.size(), v2.size());

        if(v1.size() > 0)    l.push_back(&v1);
        if(v2.size() > 0)    l.push_back(&v2);
    }

    int next() 
    {
        int ret = (*l[i])[x];
        while(moveon(), x < max_x && l[i]->size() <= x);return ret;
　　　　 return ret;
    }

    bool hasNext() 
    {
        return x < max_x;
    }
};