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

题目描述

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

解题思路

主要判断是属于单数层还是偶数层，设置一个统计层数的count，单数层采用层次遍历从左边到右边，偶数层采用从右边到左边。

代码块

import java.util.ArrayList;
import java.util.Stack;
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
         Stack<TreeNode> s1=new Stack<TreeNode>();
		 Stack<TreeNode> s2=new Stack<TreeNode>();
		 ArrayList<ArrayList<Integer> > res = new ArrayList<ArrayList<Integer> >();
		 TreeNode p=null;
		 if(pRoot==null) {
			 return res;
		 }
		 int count=1;
		 ArrayList<Integer> temp=new ArrayList<Integer>();
		 s1.add(pRoot);
        
		 while(!s1.isEmpty() || !s2.isEmpty()) {
			 if(count%2!=0) {
				 while(!s1.isEmpty()) {
					p=s1.pop();
					temp.add(p.val);
					if(p.left!=null) {
						s2.add(p.left);
					}
					if(p.right!=null) {
						s2.add(p.right);
					}
					 
				 } 
			 }
			 if(count%2==0) {
				 while(!s2.isEmpty()) {
					p=s2.pop();
					temp.add(p.val);
					if(p.right!=null) {
						s1.add(p.right);
					}
					if(p.left!=null) {
						s1.add(p.left);
					}
					 
				 }
				 
			 }
			     count++;
                 res.add(new ArrayList<Integer>(temp));
                 temp.clear(); 
		    
		 }
		 
		 return res;
	
    }

}