码迷,mamicode.com
首页 > 其他好文 > 详细

341. Flatten Nested List Iterator

时间:2018-10-31 13:51:36      阅读:189      评论:0      收藏:0      [点我收藏+]

标签:--   asn   ems   .com   list   pre   ide   imp   ack   

https://leetcode.com/problems/flatten-nested-list-iterator/discuss/80147/Simple-Java-solution-using-a-stack-with-explanation

 

 1 public class NestedIterator implements Iterator<Integer> {
 2 
 3     Stack<NestedInteger> stack = new Stack<>();
 4     public NestedIterator(List<NestedInteger> nestedList) {
 5         for(int i = nestedList.size()-1; i >= 0 ; i--){
 6             stack.push(nestedList.get(i));
 7         }
 8         
 9     }
10 
11     @Override
12     public Integer next() {
13         return stack.pop().getInteger();
14     }
15 
16     @Override
17     public boolean hasNext() {
18         while(!stack.isEmpty()){
19             NestedInteger cur = stack.pop();
20             if(cur.isInteger()){
21                 stack.push(cur);
22                 return true;
23             }
24             for(int i = cur.getList().size()-1; i >= 0; i--){
25                 stack.push(cur.getList().get(i));
26             }
27         }
28         return false;
29     }
30 }

 

341. Flatten Nested List Iterator

标签:--   asn   ems   .com   list   pre   ide   imp   ack   

原文地址:https://www.cnblogs.com/goPanama/p/9882302.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!