标签:elements hasNext helper list LeetCode341 List next 链表 Flatten
Given a nested list of integers, implement an iterator to flatten it.
Each element is either an integer, or a list -- whose elements may also be integers or other lists.
Example 1:
Input: [[1,1],2,[1,1]]
Output: [1,1,2,1,1]
Explanation: By calling next repeatedly until hasNext returns false,
the order of elements returned by next should be: [1,1,2,1,1]
.
Example 2:
Input: [1,[4,[6]]]
Output: [1,4,6]
Explanation: By calling next repeatedly until hasNext returns false,
the order of elements returned by next should be: [1,4,6].
这道题的关键在于理解输入。
本题的输入是一个嵌套结构,每个元素可能只是一个整数,也可能是一个链表;而链表中也可能再包含链表结构。面对这种输入,递归常常是一个方便的解决方案。
class NestedIterator {
private:
list<int> l;
void helper(vector<NestedInteger> input) {
for (auto it : input) {
if (it.isInteger()) l.push_back(it.getInteger());
else helper(it.getList());
}
}
public:
NestedIterator(vector<NestedInteger>& nestedList) {
helper(nestedList);
}
int next() {
int ans = l.front();
l.pop_front();
return ans;
}
bool hasNext() {
return !l.empty();
}
};
标签:elements,hasNext,helper,list,LeetCode341,List,next,链表,Flatten 来源: https://blog.csdn.net/m0_37518259/article/details/100627660
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。