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

最新推荐文章于 2021-12-29 15:16:28 发布

原创最新推荐文章于 2021-12-29 15:16:28 发布 · 373 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

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

数据结构和算法专栏收录该内容

75 篇文章

订阅专栏

本文介绍了一种特殊的二叉树遍历方法——之字形打印，即按层打印但方向交替变化。通过使用两个栈来区分奇数层和偶数层节点，实现了从左至右再从右至左的打印顺序。

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

1.本题知识点

二叉树，栈

2. 题目描述

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

3. 思路

关键点：用两个栈分别保存奇数层和偶数层的节点。

在这里插入图片描述

如上图，打印顺序及过程如下：

初始化，将1放入stack1.

1 奇数层 ---- stack1弹出1打印，并将1的左右子节点放入stack2.

3 2 偶数层 ---- stack2弹出2,3，并将左右子节点放入stack1.

4 5 6 7 奇数层 ---- stack1弹出，并将左右子节点放入stack2.

15 14 13 12 12 10 9 8 偶数层 ---- stack2弹出，…

Java版本：

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) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();//final
        Stack<TreeNode> s1 = new Stack<>();//奇数层栈
        s1.push(pRoot);//初始化
        Stack<TreeNode> s2 = new Stack<>();//偶数层栈
        int layer = 1; //记录奇数偶数的标识
        
        while(!s1.isEmpty() || !s2.isEmpty()){
            //保存每层节点数据
            ArrayList<Integer> layerList = new ArrayList<>();
            //如果是奇数层
            if(layer % 2 != 0){
                //遍历奇数栈中每一个节点，打印同时将其子节点放入偶数栈中
                while(!s1.isEmpty()){
                    TreeNode node = s1.pop();
                    if(node != null){
                        layerList.add(node.val);//打印
                        //将左右子树加入偶数栈
                        s2.push(node.left);
                        s2.push(node.right);
                    }
                }
                //当前层放入最终列表中
                if(!layerList.isEmpty()){
                    result.add(layerList);
                }
                
                layer++;//处理完一层后，改变奇偶标识符
            }
            //如果是偶数层
            else{
                //遍历偶数栈中每一个节点，打印同时将其子节点放入奇数栈中
                while(!s2.isEmpty()){
                    TreeNode node = s2.pop();
                    if(node != null){
                        layerList.add(node.val);//打印
                        //将左右子树加入奇数栈
                        //由于奇数栈打印是从左往右，所以奇数栈入栈要从右往左
                        s1.push(node.right);
                        s1.push(node.left);
                    }
                }
                //当前层放入最终列表中
                if(!layerList.isEmpty()){
                    result.add(layerList);
                }
                
                layer++;//处理完一层后，改变奇偶标识符
            }
        }
        
        return result;
    }
}