首页 > 其他 > 详细

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

时间:2017-08-18 14:04:17      阅读:325      评论:0      收藏:0      [点我收藏+]

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.

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!