在C ++中展平2D矢量
假设我们有一个2D向量，我们必须设计和实现一个迭代器来平铺2D向量。会有以下不同的方法-
next()-这将返回当前元素的下一个元素
hasNext()-这将检查下一个元素是否存在
因此，如果输入类似于[[1,2,,[3]，[4]]，则我们按如下方式调用函数-
iterator.next();
iterator.next();
iterator.next();
iterator.hasNext();
iterator.hasNext();
iterator.next();
iterator.hasNext();
那么输出将是[1,2,3，true，true，4，false]
为了解决这个问题，我们将遵循以下步骤-
定义一个2D数组v
定义初始值设定项，它将采用一个2D数组v
rowPointer：=0
colPointer：=0
n：=v的大小
而（rowPointer<n和colPointer>=v[rowPointer]的大小），则执行-
（将rowPointer增加1）
定义功能next()
x：=v[rowPointer，colPointer]
（使colPointer增加1）
如果colPointer与v[rowPointer]的大小相同，则-
（将rowPointer增加1）
colPointer：=0
（将rowPointer增加1）
而（rowPointer<n和colPointer>=v[rowPointer]的大小），则执行-
返回x
定义功能hasNext()
当rowPointer与n相同时返回false
例
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h>
using namespace std;
class Vector2D {
public:
int rowPointer, colPointer;
int n;
vector<vector<int< > v;
Vector2D(vector<vector<int< >& v){
this->v = v;
rowPointer = 0;
colPointer = 0;
n = v.size();
while (rowPointer < n && colPointer >= v[rowPointer].size()){
rowPointer++;
}
}
int next(){
//cout << rowPointer << " " << colPointer << endl;
int x = v[rowPointer][colPointer];
colPointer++;
if (colPointer == v[rowPointer].size()) {
colPointer = 0;
rowPointer++;
while (rowPointer < n && colPointer >= v[rowPointer].size()) {
rowPointer++;
}
}
return x;
}
bool hasNext(){
return !(rowPointer == n);
}
};
main(){
vector<vector<int<> v = {{1,2},{3},{4}};
Vector2D ob(v);
cout << (ob.next()) << endl;
cout << (ob.next()) << endl;
cout << (ob.next()) << endl;
cout << (ob.hasNext()) << endl;
cout << (ob.next()) << endl;
cout << (ob.hasNext());
}输入值
ob.next() ob.next() ob.next() ob.hasNext() ob.next() ob.hasNext()
输出结果
1 2 3 1 4 0