LintCode 601: Flatten 2D Vector (iterator经典题)

  1. Flatten 2D Vector

Implement an iterator to flatten a 2d vector.

Example
Example 1:

Input:[[1,2],[3],[4,5,6]]
Output:[1,2,3,4,5,6]
Example 2:

Input:[[7,9],[5]]
Output:[7,9,5]

注意这题不能用一个2D vector的copy,不然空间不满足要求。
解法1:利用C++的iterator
这题学到了很多iterator的知识。

  1. 一维和二维vector都有iterator,一维vector的iterator就好比指向每个元素的指针,二维vector的iterator就好比指向其中每个一维vector的指针。
  2. 空的一维和二维vector的begin()和end()相等。
  3. *(iter++)先返回*iter,然后iter++。
    跟*(i++)一回事。
  4. iter像指针,但不能与指针比较。一个未初始化的iter不能与NULL比较。如果非要初始化,可将其初始化为对应container的end()。
    即vector a = {1,2,3,4,5};
    vector::iterator iter = a.end();

代码如下:

class Vector2D {
public:
    Vector2D(vector<vector<int>>& vec2d) {
        begin = vec2d.begin();
        end = vec2d.end();
        if (begin != end) iter = (*begin).begin(); //注意如果begin是空指针,那么(*begin).begin()就出错。必须加if检查。
    }

    int next() {
        hasNext();
        return *(iter++);
    }

    bool hasNext() {
        while(begin != end && (iter == (*begin).end())) {
            begin++;
            iter = (*begin).begin();
        }
        return begin != end;
    }

private:
    vector<vector<int>>::iterator begin, end;
    vector<int>::iterator iter;
};

/**
 * Your Vector2D object will be instantiated and called as such:
 * Vector2D i(vec2d);
 * while (i.hasNext()) cout << i.next();
 */

二刷:

class Vector2D {
public:
    Vector2D(vector<vector<int>>& vec2d) {
        rowIter = vec2d.begin();
        vec2dEnd = vec2d.end();
        if (rowIter != vec2dEnd) colIter = rowIter->begin();
    }

    int next() {
        hasNext();
        return *colIter++;
    }

    bool hasNext() {
        while (rowIter != vec2dEnd && colIter == rowIter->end()) {
            rowIter++;
            colIter = rowIter->begin();
        }
        return rowIter != vec2dEnd;
    }
private:
    vector<vector<int>>::iterator rowIter, vec2dEnd;
    vector<int>::iterator colIter;
};

三刷: 代码不是很精炼。注意测试程序会在每次call next()之前调用hasNext()。如果hasNext()返回false,测试程序就不会前进了。

class Vector2D {
public:
    Vector2D(vector<vector<int>>& vec2d) : data(vec2d) {
        // Initialize your data structure here
        x = 0;
        y = 0;
    }

    int next() {
        if (hasNext()) {
            int res = data[x][y];
            y++;
            return res;
        }
    }

    bool hasNext() {
        if (x == data.size()) return false;
        if (y == data[x].size()) {
            y = 0;
            x++;
        }
        if (x == data.size() - 1 && data[x].size() == 0) return false;        
        return x < data.size();
    }
private:
    vector<vector<int>> data;
    int x, y;
};

/**
 * Your Vector2D object will be instantiated and called as such:
 * Vector2D i(vec2d);
 * while (i.hasNext()) cout << i.next();
 */

四刷:参考的网上的做法。先把vec2d拷贝到vec1d。注意关键是当vec2d[i]为空时,是不会copy任何东西到vec1d的,这样就避开了vec2d里面的空数组的坑。
vec1d.insert(vec1d.end(), vec2d[i].begin(), vec2d[i].end());

class Vector2D {
public:
    Vector2D(vector<vector<int>>& vec2d) {
        for (int i = 0; i < vec2d.size(); i++) {
            vec1d.insert(vec1d.end(), vec2d[i].begin(), vec2d[i].end());
        }
    }
    int next() {
        return vec1d[index++];
    }
    bool hasNext() {
        return index < vec1d.size();
    }
private:
    vector<int> vec1d;
    int index = 0;
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值