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

[GeeksForGeeks] Level order traversal in spiral form of a binary tree.

时间:2017-08-18 14:30:31      阅读:136      评论:0      收藏:0      [点我收藏+]

标签:des   lag   code   evel   bin   cti   direct   sed   logs   

Write a function to print spiral order traversal of a binary tree. For below tree, function should print 1, 2, 3, 4, 5, 6, 7.


 技术分享 

 


Solution.

For a normal level order traversal of a binary tree, we traverse every level from left to right. To achieve this, a queue 

is used to store the next level‘s nodes. 

To do this traversal in a spiral order, we need to be able to add and poll nodes in a queue from its front and back.

This implies using Deque and a boolean flag to indicate if we are traversing from left to right or from right to left.

When traversing from right to left, we poll nodes from the end of dq; First add its right child node, then its left child node to the front of dq.

When traversing from left to right, we poll nodes from the front of dq; First add its left, then right child node to the back of dq.

After adding all nodes in one level, reverse the traversal direction flag.

 

 1 import java.util.ArrayList;
 2 import java.util.LinkedList;
 3 import java.util.Deque;
 4 
 5 class LOTSpiralNode {
 6     int val;
 7     LOTSpiralNode left, right;
 8     LOTSpiralNode(int v) {
 9         this.val = v;
10         this.left = null;
11         this.right = null;
12     }
13     LOTSpiralNode(LOTSpiralNode t){
14         this.val = t.val;
15         this.left = t.left;
16         this.right = t.right;
17     }
18 }
19 public class LOTSpiral {
20     public ArrayList<Integer> levelOrderTraversalSpiral(LOTSpiralNode root) {
21         ArrayList<Integer> list = new ArrayList<Integer>();
22         if(root == null) {
23             return list;
24         }
25         Deque<LOTSpiralNode> dq = new LinkedList<LOTSpiralNode>();
26         boolean leftToRight = false;
27         dq.add(root);
28         while(!dq.isEmpty()) {
29             int size = dq.size();
30             for(int i = 0; i < size; i++) {
31                 LOTSpiralNode curr;
32                 if(leftToRight) {
33                     curr = dq.pollFirst();
34                     if(curr.left != null) {
35                         dq.addLast(curr.left);
36                     }
37                     if(curr.right != null) {
38                         dq.addLast(curr.right);
39                     }
40                 }
41                 else{
42                     curr = dq.pollLast();
43                     if(curr.right != null) {
44                         dq.addFirst(curr.right);
45                     }
46                     if(curr.left != null) {
47                         dq.addFirst(curr.left);
48                     }
49                 }
50                 list.add(curr.val);
51             }
52             leftToRight = !leftToRight;
53         }
54         return list;
55     }
56 }

 

[GeeksForGeeks] Level order traversal in spiral form of a binary tree.

标签:des   lag   code   evel   bin   cti   direct   sed   logs   

原文地址:http://www.cnblogs.com/lz87/p/7349794.html

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