剑指offer 之字形打印二叉树 java实现

最新推荐文章于 2024-07-14 14:13:59 发布

leoma2012

最新推荐文章于 2024-07-14 14:13:59 发布

阅读量401

点赞数

CC 4.0 BY-SA版权

文章标签：剑指offer

本文链接：https://blog.youkuaiyun.com/leoma2012/article/details/88796170

本文介绍了一种二叉树的层次遍历方法，并通过两个栈实现按从上到下、从左到右的顺序输出节点值。对于奇数层，从左到右输出；对于偶数层，从右到左输出。此算法使用了两个栈，分别存储奇数层和偶数层的节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码如下

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<>();//存放奇数层节点，从第一层开始计算
        Stack<TreeNode> s2 = new Stack<>();//存放偶数层节点
        ArrayList< ArrayList<Integer> > ans = new ArrayList<>();
        if(pRoot == null)
            return ans;
        s1.push(pRoot);
        while( !s1.empty() || !s2.empty() ){
            ArrayList<Integer> temp = new ArrayList<>();//临时结果
            if(s2.empty())
                while(!s1.empty()){
                    //奇数层的子节点弹出顺序是从右到左，则存放顺序是从左到右
                    TreeNode top = s1.pop();
                    temp.add(top.val);
                    if(top.left != null)
                        s2.push(top.left);
                    if(top.right != null)
                        s2.push(top.right);
                }
            else
                while(!s2.empty()){
                    //与奇数层正好相反
                    TreeNode top = s2.pop();
                    temp.add(top.val);
                    if(top.right != null)
                        s1.push(top.right);
                    if(top.left != null)
                        s1.push(top.left);
                }
            ans.add(temp);
        }
        return ans;
    }
}