[Leetcode] 251. Flatten 2D Vector 解题报告

最新推荐文章于 2025-10-13 11:50:44 发布

原创最新推荐文章于 2025-10-13 11:50:44 发布 · 740 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode #解题报告 #Design

IT公司面试习题专栏收录该内容

736 篇文章

订阅专栏

本文介绍了一种实现2D向量迭代器的方法，通过维护动态迭代器row_it和col_it来遍历二维数组，并确保能正确处理空行情况。提供了一个C++示例，演示如何使用迭代器按顺序访问所有元素。

题目：

Implement an iterator to flatten a 2d vector.

For example,
Given 2d vector =

[
  [1,2],
  [3],
  [4,5,6]
]

By calling next repeatedly until hasNext returns false, the order of elements returned by next should be: [1,2,3,4,5,6].

Follow up:
As an added challenge, try to code it using only iterators in C++ or iterators in Java.

思路：

需要维护两个动态迭代器row_it和col_it，分别表示当前位置的行和列。但是由于我们不存储vec2d本身，所以我们还需要另外一个静态迭代器row_end，表示整个二维数组的末尾。这里需要特别注意的是在判断hasNext的时候，要处理那种跳过空行的情况，请见下面的代码实现。

代码：

class Vector2D {
public:
    Vector2D(vector<vector<int>>& vec2d) {
        if (vec2d.size() == 0) {
            return;
        }
        row_it = vec2d.begin(), row_end = vec2d.end();
        col_it = (*row_it).begin();
    }

    int next() {
        return *col_it++;
    }

    bool hasNext() {
        while (row_it != row_end && col_it == (*row_it).end()) {
            ++row_it;
            col_it = (*row_it).begin();
        }
        return row_it != row_end;
    }
private:
    vector<vector<int>>::iterator row_it, row_end;
    vector<int>::iterator col_it;
};

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