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

59按之字形顺序打印二叉树

时间:2018-01-12 21:15:13      阅读:191      评论:0      收藏:0      [点我收藏+]

标签:描述   nbsp   col   pop   push   打印   arraylist   二叉树   treenode   

题目描述

请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。

需要2个桟(后进先出)。
我们在打印某一行节点时,把下一层的子节点保存到相应的桟里。如果当前打印的是奇数层,则先保存左子树再保存右子树节点到
第一个桟里,如果当前打印的是偶数层,则先保存右子树再保存左子树节点。

 1 import java.util.ArrayList;
 2 import java.util.Stack;
 3 /*
 4 public class TreeNode {
 5     int val = 0;
 6     TreeNode left = null;
 7     TreeNode right = null;
 8 
 9     public TreeNode(int val) {
10         this.val = val;
11 
12     }
13 
14 }
15 */
16 public class Solution {
17     public ArrayList<ArrayList<Integer> > Print(TreeNode root) {
18         ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
19         Stack<TreeNode> stack1 = new Stack<TreeNode>();
20         Stack<TreeNode> stack2 = new Stack<TreeNode>();
21         int flag = 1;
22         stack1.push(root);
23         while(!stack1.empty()||!stack2.empty()){
24             if(flag==1){
25                 ArrayList<Integer> res_temp = new ArrayList<Integer>(); 
26                 while(!stack1.empty()){
27                     TreeNode resNode = stack1.pop();
28                     if(resNode !=null){
29                         res_temp.add(resNode.val);
30                         stack2.push(resNode.left);
31                         stack2.push(resNode.right);
32                     }
33                 }
34                 if(!res_temp.isEmpty())
35                     res.add(res_temp);
36                 flag = 1-flag;
37             }
38             else{
39                 ArrayList<Integer> res_temp = new ArrayList<Integer>(); 
40                 while(!stack2.empty()){
41                     TreeNode resNode = stack2.pop();
42                     if(resNode!=null){
43                         res_temp.add(resNode.val);
44                         stack1.push(resNode.right);
45                         stack1.push(resNode.left);
46                     }
47                 }
48                    if(!res_temp.isEmpty())
49                     res.add(res_temp);
50                  flag = 1-flag;
51             }
52         }
53         return res;
54     }
55 
56 }

 

59按之字形顺序打印二叉树

标签:描述   nbsp   col   pop   push   打印   arraylist   二叉树   treenode   

原文地址:https://www.cnblogs.com/zle1992/p/8277596.html

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