按之字形順序打印二叉樹(用棧分奇偶行不同方向添加打印)

題目描述

請實現一個函數按照之字形打印二叉樹，即第一行按照從左到右的順序打印，第二層按照從右至左的順序打印，第三行按照從左到右的順序打印，其他行以此類推。

 

分層處理：利用兩個棧，一個棧存儲奇數層的節點，一個棧用來存儲偶數層的節點

當遍歷到奇數層節點時，利用上一偶數層棧中的節點向奇數層棧添加節點

當遍歷到偶數層節點時，利用上一奇數層棧中的節點向偶數層棧添加節點

注意添加時的方向不同

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;
	}
}