剑指offer——按之字形顺序打印二叉树

问题描述:

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


思路:

之字形打印其实与上一个按层次打印基本相同,唯一不同的是每一层的顺序不一样,所以在这里可以定义一个变量,打印一层之后将其与1异或,作为打印顺序的标志位。


代码实现:

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

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
public class Solution {
    public ArrayList > Print(TreeNode pRoot) {
        ArrayList> resultList = new ArrayList>();      
        if(pRoot==null) return resultList;
        //创建一个队列用于保存需要打印的结点
        Queue nodeQueue = new LinkedList();
        int nextNum =0; //本层已经打印的数量
        int toBePrint = 1; //本层需要打印的总数量
        int isLeftToRight = 0;//用来判断方向
        nodeQueue.add(pRoot);

        ArrayList result = new ArrayList();

        while(!nodeQueue.isEmpty()){
            TreeNode curNode = nodeQueue.poll();
            result.add(curNode.val);
            nextNum++;
            if(curNode.left!=null) nodeQueue.add(curNode.left);
            if(curNode.right!=null) nodeQueue.add(curNode.right);
            //判断本层是否打印完毕
            if(nextNum == toBePrint){
                if(isLeftToRight==1){
                    resultList.add(reverse(result));
                }else{
                    resultList.add(result);
                }
                toBePrint = nodeQueue.size();
                nextNum = 0;
                result = new ArrayList();
                isLeftToRight = isLeftToRight^1;
            }
        }
        return resultList;
    }
    private ArrayList reverse(ArrayList result){
        ArrayList reverseResult = new ArrayList();
        for(int i=result.size()-1;i>=0;i--){
            reverseResult.add(result.get(i));
        }
        return reverseResult;
    }

}

你可能感兴趣的:(剑指offer)