剑指offer之打印二叉树

题目一　

（一）题目：从上往下打印出二叉树的每个结点，同一层的结点按照从左到右的顺序打印。

（二）思路：不分行从上往下打印二叉树：该题即为对二叉树的层序遍历，结点满足先进先出的原则，采用队列。每从队列中取出头部结点并打印，若其有子结点，把子结点放入队列尾部，直到所有结点打印完毕。

（三）代码：

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

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

    }

}
*/
public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> list=new  ArrayList<Integer>();
         ArrayList<TreeNode> queue=new  ArrayList<TreeNode>();
        if(root==null)
            return list;
        
        queue.add(root);
        while(queue.size()!=0){
            TreeNode node=queue.remove(0);
         
            
            if(node.left!=null){
                queue.add(node.left);
            }
            if(node.right!=null){
                queue.add(node.right);
            }
               list.add(node.val);
        }
        return list;
    }
}

题目二　

（一）题目：从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

（二）思路：同样使用队列，但比第一题增加两个变量：当前层结点数目pCount，下一层结点数目nextCount。根据当前层结点数目来打印当前层结点，同时计算下一层结点数目，之后令pCount等于nextCount，重复循环，直到打印完毕。

（三）代码：

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 {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();//存放结果
        if(pRoot == null){
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();//使用队列保存将要打印的结点，先进先出
        ArrayList<Integer> arrayList = new ArrayList<Integer>();//存放每行数据的列表
        queue.offer(pRoot);
        int pCount= 0;  //记录本层已经打印了多少个
        int nextCount = 1;//记录下层应该打几个
        while(!queue.isEmpty()){
           TreeNode cur = queue.poll();
           arrayList.add(cur.val);//添加到本行的arrayList
            pCount++;//打印一个，计数加一，直到全部打印完，再继续从0计数
            //每打印一个节点，就把此节点的下一层的左右节点加入队列，并记录下一层要打印的个数
            if(cur.left!=null){
               queue.offer(cur.left);           
            }
            if(cur.right!=null){
               queue.offer(cur.right);
            }
              //判断本层打印是否完成
            if(pCount == nextCount){
                nextCount = queue.size();   //此时的queue中存储的都是下一层的节点，
                pCount = 0;//从头开始打印下一行
                result.add(arrayList);//把arrayList添加到结果列表中
                arrayList = new ArrayList<Integer>();  //重置arrayList
            }
        }
        return result;
    }
}

题目三　

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

（二）思路：采用两个栈（分别用于存储奇数和偶数层的结点）

                        打印偶数层结点（先右出再左出）：先保存左子结点在保存右子结点

                        打印奇数层结点（先左出再右出）：先保存右子结点在保存左子结点

（三）代码：

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 {
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
 
        ArrayList<ArrayList<Integer> > listAll = new ArrayList<>();
        if(pRoot==null)return listAll;
        
        Stack<TreeNode> s1 = new Stack<>();//s1存奇数层结点
        Stack<TreeNode> s2 = new Stack<>();//s2存偶数层结点
        
        int level = 1;//判断层数
        s1.push(pRoot);
        
        while(!s1.isEmpty()||!s2.isEmpty()){
            ArrayList<Integer> list = new ArrayList<>();
            if(level++%2!=0){//奇数层，最开始先弹出根结点，放入结果链表中，再把左右子树的结点放入栈中
                while(!s1.isEmpty()){
                    TreeNode node = s1.pop();
                    list.add(node.val);
                    if(node.left!=null)s2.push(node.left);
                    if(node.right!=null)s2.push(node.right);
                }
            }
            else{
                while(!s2.isEmpty()){
                    TreeNode node = s2.pop();
                    list.add(node.val);
                    if(node.right!=null)s1.push(node.right);
                    if(node.left!=null)s1.push(node.left);
                }
            }
            listAll.add(list);
        }
        return listAll;
   }
}