黄视频网站在线观看,久草免费在线视频,五月激情偷拍网,特级丰满少妇一级AAAA爱毛片,色色一区二区,日本岛国视频在线观看一区二区三区 ,日韩一级二级三级片,看肏屄

算法的重要性，我就不多說了吧，想去大廠，就必須要經(jīng)過基礎(chǔ)知識和業(yè)務(wù)邏輯面試+算法面試。所以，為了提高大家的算法能力，這個公眾號后續(xù)每天帶大家做一道算法題，題目就從LeetCode上面選！

今天和大家聊的問題叫做 展開二維向量，我們先來看題面：

https://leetcode-cn.com/problems/flatten-2d-vector/

Implement an iterator to flatten a 2d vector.

請設(shè)計并實現(xiàn)一個能夠展開二維向量的迭代器。該迭代器需要支持 next 和 hasNext 兩種操作。

示例

Input: 2d vector =
[
  [1,2],
  [3],
  [4,5,6]
]
Output: [1,2,3,4,5,6]

解題

需要維護(hù)兩個動態(tài)迭代器row_it和col_it，分別表示當(dāng)前位置的行和列。但是由于我們不存儲vec2d本身，所以我們還需要另外一個靜態(tài)迭代器row_end，表示整個二維數(shù)組的末尾。這里需要特別注意的是在判斷hasNext的時候，要處理那種跳過空行的情況，請見下面的代碼實現(xiàn)。

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