按之字形顺序打印二叉树(用栈分奇偶行不同方向添加打印)

题目描述

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

 

分层处理：利用两个栈，一个栈存储奇数层的节点，一个栈用来存储偶数层的节点

当遍历到奇数层节点时，利用上一偶数层栈中的节点向奇数层栈添加节点

当遍历到偶数层节点时，利用上一奇数层栈中的节点向偶数层栈添加节点

注意添加时的方向不同

import java.util.ArrayList;
import java.util.Stack;


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) {
		ArrayList<ArrayList<Integer>> res = new ArrayList<>();
		int layer = 1;
		Stack<TreeNode> s1 = new Stack<TreeNode>();
		s1.push(pRoot);
		Stack<TreeNode> s2 = new Stack<TreeNode>();

		while (!s1.empty() || !s2.empty()) {
			if (layer % 2 == 1) {// 奇数层
				ArrayList<Integer> value = new ArrayList<>();
				while (!s1.isEmpty()) {
					TreeNode node = s1.pop();
					if (node != null) {
						value.add(node.val);
						s2.push(node.left);
						s2.push(node.right);
					}
				}
				if(!value.isEmpty()){
					res.add(value);
					layer++;
				}
			}else{
				ArrayList<Integer> value = new ArrayList<>();
				while (!s2.isEmpty()) {
					TreeNode node = s2.pop();
					if (node != null) {
						value.add(node.val);
						s1.push(node.right);
						s1.push(node.left);
					}
				}
				if(!value.isEmpty()){
					res.add(value);
					layer++;
				}
			}
		}
		return res;
	}
}