剑指offer系列之58:按之字形顺序打印二叉树

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

思路:此题明显是层序遍历的思路。由于需要按照之字形打印每一层,相当于在打印每一行之前需要判断上一行打印的顺序。比如,如果上一行打印的顺序使从左到右,那么下一行的打印顺序应该是从右到左。实现的这点可以采用奇数行从左到右打印,偶数行从右到左进行打印。还可直接设置 一个布尔变量,每打印一行就改变一次该变量的值。其他的就是层序遍历的思路了。

import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;

/*
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>> resultList = new ArrayList<ArrayList<Integer>>();
        if(pRoot == null) return resultList;
        //创建一个队列保存遍历的顺序
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(pRoot);
        //创建每一行遍历的结果
        ArrayList<Integer> result = new ArrayList<Integer>();
        boolean isLeftToRight = true;
        int begin = 0,end = 1;
        while(!queue.isEmpty()){
            TreeNode curNode = queue.poll();
            result.add(curNode.val);
            begin++;
            if(curNode.left != null) queue.add(curNode.left);
            if(curNode.right != null) queue.add(curNode.right);
            //判断一行的所有元素是否都添加到result中
            if(begin == end){
                //判断当前行按照怎样的顺序打印
                if(!isLeftToRight){
                    resultList.add(reverse(result));
                }else{
                    resultList.add(result);
                }
                //下一行需要改变方向
                isLeftToRight = !isLeftToRight;
                end = queue.size();
                begin = 0;
                //一行打印完毕之后,需要创建的result集合保存下一行遍历的元素
                result = new ArrayList<Integer>();
            }
        }
        return resultList;
    }

    private ArrayList<Integer> reverse(ArrayList<Integer> result) {
        int size = result.size();
        ArrayList<Integer> reverseResult = new ArrayList<Integer>();
        for(int i = size - 1; i >= 0; i--){
            reverseResult.add(result.get(i));
        }
        return reverseResult;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值