按之字形顺序打印二叉树

最新推荐文章于 2023-08-16 22:20:52 发布

wenlyq

最新推荐文章于 2023-08-16 22:20:52 发布

阅读量91

点赞数

CC 4.0 BY-SA版权

分类专栏：二叉树剑指offer

本文链接：https://blog.youkuaiyun.com/wenlyq/article/details/88929769

剑指offer 同时被 2 个专栏收录

57 篇文章

订阅专栏

二叉树

18 篇文章

订阅专栏

本文介绍了一种特殊的二叉树打印方式——之字形打印，并提供了两种实现方案。一种是通过层次遍历结合队列的方法，另一种是使用两个栈来提高效率。

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

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

思路：利用层次遍历，但加入本层levelque和下一层nextlevel，加入righttoleft判断是否需要逆序，res存放最终值
1.根节点加入levelque中
2.while leveque：针对每层创建一个nextlevel（存入左子树和右子树），curvalue存放当前层的值
3.根据是否需要逆序，将curvalue是否逆序
4.然后将curvalue加入res中，nextlevel给level

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def Print(self, pRoot):
        # write code here
        if not pRoot:
            return []
        levelque=[pRoot]
        righttoleft=False
        res=[]
        while levelque:
            curvalue=[]
            nextlevel=[]
            for i in levelque:
                curvalue.append(i.val)
                if  i.left:
                    nextlevel.append(i.left)
                if  i.right:
                    nextlevel.append(i.right)
            if righttoleft:
                curvalue.reverse()
            if curvalue:
                res.append(curvalue)
            levelque=nextlevel
            righttoleft= not righttoleft
        return res

reverse方法时间复杂度比较高

思路二：用两个stack，用空间换取时间效率

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>> res=new ArrayList<>();
        if(pRoot==null) return res;
        Stack<TreeNode> s1=new Stack<>();
        Stack<TreeNode> s2=new Stack<>();
        s1.push(pRoot);
        int level=1;
        while(!s1.isEmpty()||!s2.isEmpty()){
            ArrayList<Integer> list=new ArrayList<>();
            if(level++%2!=0){//层级，奇数，偶数
                while(!s1.isEmpty()){
                    TreeNode cur=s1.pop();
                    list.add(cur.val);
                    if(cur.left!=null) s2.push(cur.left);//s2逆序，先进后厨
                    if(cur.right!=null) s2.push(cur.right);
                }
            }
            else{//逆序
                while(!s2.isEmpty()){
                    TreeNode cur=s2.pop();
                    list.add(cur.val);
                    if(cur.right!=null) s1.push(cur.right);
                    if(cur.left!=null) s1.push(cur.left);
                }
            }
            res.add(list);
        }
        return res;
    }
}