JAVA 按之字形顺序打印二叉树

最新推荐文章于 2024-08-22 10:44:26 发布

原创最新推荐文章于 2024-08-22 10:44:26 发布 · 868 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#java #二叉树 #数据结构

java基础专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种特殊的二叉树遍历方法——之字形打印。通过使用两个栈来区分奇数行和偶数行，实现了从左到右及从右到左交替打印节点值的功能。代码示例中详细展示了如何构建一个完全二叉树并进行之字形打印。

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

分别用两个栈存放树的奇数行和偶数行，奇数行自左向右压入孩子，偶数行自右向左压入孩子。

import java.util.LinkedList;
import java.util.Stack;

/**
 * Created by Administrator on 2018/3/7.
 */
public class zhiPrint {
    public void print(TreeNode root){
        if(root==null)
            return;
        Stack<TreeNode> s1 = new Stack<>();//存放奇数行
        Stack<TreeNode> s2 = new Stack<>();//存放偶数行
        s1.push(root);
        while(!s1.isEmpty() || !s2.isEmpty()){
            while (!s1.isEmpty()) {//弹出奇数行所有元素
                TreeNode node = s1.pop();
                System.out.print(node.element+" ");
                if (s1.isEmpty())//判断该行是否结束
                    System.out.println();
                if (node.left != null)
                    s2.push(node.left);//先左child再右child压入偶数行的栈
                if (node.right != null)
                    s2.push(node.right);
            }
            while (!s2.isEmpty()) {//弹出偶数行所有元素
                TreeNode node = s2.pop();
                System.out.print(node.element+" ");
                if (s2.isEmpty())
                    System.out.println();
                if (node.right != null)
                    s1.push(node.right);//先右child再左child压入奇数行的栈
                if (node.left != null)
                    s1.push(node.left);
            }
        }
    }
    public static void main(String... args){
        LinkedList<TreeNode> q = new LinkedList<>();//用队列构建一棵完全二叉树，节点从1至15
        int i = 1;
        TreeNode root = new TreeNode(i++);
        q.add(root);
        while (!q.isEmpty()&&i<=15){
            TreeNode lchild = new TreeNode(i++);
            TreeNode rchild = new TreeNode(i++);
            TreeNode node = q.poll();
            node.left = lchild;
            node.right = rchild;
            q.add(node.left);
            q.add(node.right);
        }
        //之字打印
        new zhiPrint().print(root);
    }
}
class TreeNode {
    Object element;
    TreeNode left;
    TreeNode right;
    public TreeNode(int id){
        element = (Object)id;
    }
}