码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode "Zigzag Iterator"

时间:2015-09-15 06:56:31      阅读:152      评论:0      收藏:0      [点我收藏+]

标签:

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; } };

LeetCode "Zigzag Iterator"

标签:

原文地址:http://www.cnblogs.com/tonix/p/4809038.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!