【注释详细,思路清晰】《剑指Offer》:按之字型顺序打印二叉树

1、题目描述

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

2、题目分析

① 之字型打印不如是  Z字型打印。首先想到的是用一个栈,用一个队列。但是使用两个栈更加的方便,一个栈存储的是奇数行的元素,一个栈存储的是偶数行的元素。

② 当遍历的时候,注意当前元素的判空,以及判空后的为不空的左子树和右子树

3、代码实现

import java.util.ArrayList;
import java.util.*;
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    // 定义两个栈,stack1,stack2,stack1存储的是第一行元素,stack2存储的是第二行的元素
        // 依次类推,交替进行
    // 定义栈存储的是奇数行的元素
    private Stack<TreeNode> stackOdd = new Stack<>();
    // 定义栈存储的是偶数的元素
    private Stack<TreeNode> stackEven = new Stack<>();
    
    // 遍历二叉树,从左到右进行遍历,从左到右是true,从右到左是false
    private boolean leftToRight = true;
    
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        
        // 定义存储结果的集合容器
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        
        // 判断初始结点是否为空
        if(pRoot == null){
            return result;
        }
        
        // 将根结点入栈stackOdd
        stackOdd.push(pRoot);
        
        // 当存储奇数行的栈都不为空的时候
        while(!stackOdd.isEmpty() || !stackEven.isEmpty()){
            
            // 定义集合存储每行的元素
            ArrayList<Integer> list = new ArrayList<>();
            if(!stackOdd.isEmpty()){
                //将stackOdd中的元素添加到list中
                while(!stackOdd.isEmpty()){
                    TreeNode node = stackOdd.pop();
                    list.add(node.val);
                    // 判断当前节点的左右结点,从右往左遍历,所以进栈的时候是从左往右
                    if(node.left != null){
                        stackEven.push(node.left);
                    }
                    if(node.right != null){
                        stackEven.push(node.right);
                    }
                }
            }else{
                //奇数行添加完了,现在添加偶数行的元素
                while(!stackEven.isEmpty()){
                    TreeNode node = stackEven.pop();
                    list.add(node.val);
                    
                    //判断左右元素
                    if(node.right != null){
                        stackOdd.push(node.right);
                    }
                    if(node.left != null){
                        stackOdd.push(node.left);
                    }
                }
            }
            result.add(list);
            
        }
        return result;
        
    }

}


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值