剑指offer 59.树 按之字形顺序打印二叉树

Posted 2022-08-31 transkai

题目描述

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

 

解题思路

用一个栈s1来存放奇数层节点，另一个栈s2来存放偶数层节点

 

代码如下

public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) 
        
             int layer = 1;
             //s1存奇数层节点
             Stack<TreeNode> s1 = new Stack<TreeNode>();
             s1.push(pRoot);
             //s2存偶数层节点
             Stack<TreeNode> s2 = new Stack<TreeNode>();
              
             ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
              
             while (!s1.empty() || !s2.empty()) 
                 if (layer%2 != 0) 
                     ArrayList<Integer> temp = new ArrayList<Integer>();
                     while (!s1.empty()) 
                         TreeNode node = s1.pop();
                         if(node != null) 
                             temp.add(node.val);
                             System.out.print(node.val + " ");
                             s2.push(node.left);
                             s2.push(node.right);
                         
                     
                     if (!temp.isEmpty()) 
                         list.add(temp);
                         layer++;
                         System.out.println();
                     
                  else 
                     ArrayList<Integer> temp = new ArrayList<Integer>();
                     while (!s2.empty()) 
                         TreeNode node = s2.pop();
                         if(node != null) 
                             temp.add(node.val);
                             System.out.print(node.val + " ");
                             s1.push(node.right);
                             s1.push(node.left);
                         
                     
                     if (!temp.isEmpty()) 
                         list.add(temp);
                         layer++;
                         System.out.println();
                     
                 
             
             return list;