[leetcode]_Binary Tree Level Order Traversal I && II

4646 단어 LeetCode
제목: 두 갈래 나무 한 그루를 정하여 이를 차원적으로 훑어보고 훑어보는 결과를 2차원 체인표에 저장한다.
사고방식: 두 갈래 나무의 차원을 두루 훑어보는 관건은Queue를 사용하는 것이다.
복습:
Queue의 정의입니다.LinkedList는 Queue 인터페이스를 구현하기 때문에 LinkedList를 사용하여 Queue를 초기화할 수 있습니다.
Queue를 사용합니다.
1,offer (), 입대, 대기열이 가득 차면false로 돌아갑니다.
2,poll(), 대기열 헤더 요소를 되돌려주고 삭제합니다. 대기열이 비어 있을 때null을 되돌려줍니다.
3. peek (), 대기열 헤더 요소를 삭제하지 않고 대기열이 비어 있을 때null로 되돌려줍니다.
 
코드:
 1     public ArrayList<ArrayList<Integer>> levelOrder(TreeNode root) {
 2         
 3         ArrayList<ArrayList<Integer>> level = new ArrayList<ArrayList<Integer>>();
 4      
 5         if(root == null) return level; // , , root== null , , null。
 6         
 7         Queue<TreeNode> q = new LinkedList<TreeNode>();
 8         q.offer(root);
 9         q.offer(new TreeNode(Integer.MAX_VALUE)); // 
10         
11         ArrayList<Integer> oneLevel = new ArrayList<Integer>();
12         
13         while(!q.isEmpty()){
14             TreeNode node = q.poll();
15             
16             if(node.val != Integer.MAX_VALUE){
17                 
18                 oneLevel.add(node.val);
19                 if(node.left != null) q.offer(node.left);
20                 if(node.right != null) q.offer(node.right);
21                 
22             }else{
23                 if(oneLevel.size() == 0) break; // 
24                 
25                 level.add(oneLevel);
26                 oneLevel = new ArrayList<Integer>();
27                 q.offer(new TreeNode(Integer.MAX_VALUE));
28             
29             }
30         }
31         return level;
32     }

좋은 웹페이지 즐겨찾기