刚出炉一小时不到的Airbnb面筋，整个人都不好了。。。

又看到楼主发面经了。。。。。印象深刻的求逆序对归并排序。。。

hasNext()和next()的顺序是不是可以任意调用的？一个hasNext()后接多个next()的话，更新应该放到next()里面吧。这个感觉用迭代器而不是索引来做应该简单一些

而且做这个题的时候，一些边界情况是不是要确定好，比如说无效的next()，应该返回什么样的数据呢

简单写了点代码，不知道这样对不对

1. class 2Diterator{
   
2. public:
   
3.         2Diterator(vector<vector<int>>& vec2d){
   
4.             row = vec2d.begin();
   
5.             end = vec2d.end();
   
6.             while(row != end && (*row).empty()){
   
7.                 row++;
   
8.             }
   
9.             if(row != end)
   
10.                 col = (*row).begin();
    
11.         }
    
12.         int next(){
    
13.                 if(!hasNext()){//如果连续调用next的话，这里应该也检查一下有效性
    
14.                         return INT_MAX;//返回一个特殊值
    
15.                 }
    
16.             int res = *col;
    
17.             col++;
    
18.             getValidCol();
    
19.             return res;
    
20.         }
    
21.         bool hasNext(){
    
22.             return row != end;
    
23.         }
    
24.         void remove(){
    
25.             if(!hasNext())
    
26.                 return;
    
27.             col = (*row).erase(col);
    
28.             getValidCol();
    
29.         }
    
30. private:
    
31.                 void getValidCol(){
    
32.                         if(col == (*row).end()){
    
33.                 while(row != end){
    
34.                     row++;
    
35.                     if(row != end && !(*row).empty()){
    
36.                         col = (*row).begin();
    
37.                         return;
    
38.                     }
    
39.                 }
    
40.             }
    
41.                 }
    
42.         vector<vector<int>>::iterator row, end;
    
43.         vector<int>::iterator col;
    
44. }

复制代码