剑指 offer第32.1-3 从上往下打印二叉树

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

例如，以下二叉树层次遍历的结果为：1,2,3,4,5,6,7

使用队列来进行层次遍历，当前队列中的节点数就是当前层的节点数，只要控制遍历这么多节点数，就能保证这次遍历的都是当前层的节点。

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<Integer> PrintFromTopToBottom(TreeNode root) {
        Queue<TreeNode> queue=new LinkedList<>();
        ArrayList<Integer> ret=new ArrayList<>();
        queue.add(root);
        while(!queue.isEmpty())
        {
            int cnt=queue.size();
            while(cnt-->0)
            {
                TreeNode temp=queue.poll();
                if(temp==null){
                    continue;
                }
                ret.add(temp.val);
                queue.add(temp.left);
                queue.add(temp.right);
            }
        }
        return ret;
    }
}

Queue queue = new LinkedList<>();
LinkedList类是双向列表,列表中的每个节点都包含了对前一个和后一个元素的引用.
LinkedList的构造函数如下

1. public LinkedList(): ——生成空的链表
2. public LinkedList(Collection col): 复制构造函数

32.2 把二叉树打印成多行

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 {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
    ArrayList<ArrayList<Integer>> ret=new ArrayList<>();
        Queue<TreeNode> queue=new LinkedList<>();
        queue.add(pRoot);
        while(!queue.isEmpty()){
            ArrayList<Integer> temp=new ArrayList<>();
            int cnt=queue.size();
            while(cnt-->0){
                TreeNode node=queue.poll();
                if(node==null){
                    continue;
                }
                temp.add(node.val);
                queue.add(node.left);
                queue.add(node.right);
            }
            if(temp.size()!=0){
                ret.add(temp);
            }
        }
        return ret;
    }
    
}

添加了ArrayList<ArrayList> ret = new ArrayList<>();
ret里面是加一个list。

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

import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
import java.util.Collections;
/*
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>> ret=new ArrayList<>();
        Queue<TreeNode> queue=new LinkedList<>();
        queue.add(pRoot);
        boolean reverse=false;
        while(!queue.isEmpty())
        {
            ArrayList<Integer> temp=new ArrayList<>();
            int cnt=queue.size();
            while(cnt-->0){
                TreeNode node=queue.poll();
                if(node==null) continue;
                temp.add(node.val);
                queue.add(node.left);
                queue.add(node.right);
            }
            if(reverse)
                Collections.reverse(temp);
            reverse=!reverse;
            if(temp.size()!=0) ret.add(temp);
        }
        return ret;

    }

}

加入一个reverse来判断奇数层和偶数层，注意判断node里面的是==判断