剑指offer 面试题23 从上到下打印二叉树

剑指offer 面试题23 从上到下打印二叉树

package algorithm.foroffer.top30;

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

/**
* Created by liyazhou on 2017/5/28.
* 面试题23：从上往下打印二叉树
*
* 题目：
*      从上往下打印出二叉树的每个结点，同一层的结点按照从左到右的顺序打印。
*      例如输入下图的二叉树，则依次打印出 8、6、10、5、7、9、11.
*                  8
*                /   \
*              6      10
*            /  \    /  \
*           5   7   9   11
*
* 问题：
*      1. 层次遍历二叉树
*
* 思路：
*      1. 层次遍历二叉树，是广度优先搜索，需要使用队列
*      2. 创建队列，并把根结点入队
*         循环：队列不为空
*              队列的首元素出队，并访问该元素
*              如果当前元素的左孩子不为空，则入队
*              如果当前元素的右孩子不为空，则入队
*/

class TreeNode23{
int value;
TreeNode23 left;
TreeNode23 right;

public TreeNode23(int value){ this.value = value; }

public void setChildren(TreeNode23 left, TreeNode23 right) {
this.left = left;
this.right = right;
}
}

public class Test23 {

/**
* 层次遍历二叉树
* @param root 二叉树的根结点
*/
public static void levelVisitTree(TreeNode23 root){
if (root == null) return;

Queue<TreeNode23> queue = new LinkedList<>();
// queue.offer(root);
// while (!queue.isEmpty()){
for (queue.offer(root); !queue.isEmpty();){
TreeNode23 currNode = queue.poll();
System.out.print(currNode.value + "\t");

if (currNode.left != null) queue.offer(currNode.left);
if (currNode.right != null) queue.offer(currNode.right);
}
}

public static void main(String[] args){
TreeNode23 root = generateTree();
Test23.levelVisitTree(root);
}

private static TreeNode23 generateTree() {
TreeNode23 treeNode0 = new TreeNode23(8);

TreeNode23 treeNode1 = new TreeNode23(6);
TreeNode23 treeNode2 = new TreeNode23(10);

TreeNode23 treeNode3 = new TreeNode23(5);
TreeNode23 treeNode4 = new TreeNode23(7);
TreeNode23 treeNode5 = new TreeNode23(9);
TreeNode23 treeNode6 = new TreeNode23(11);

treeNode0.setChildren(treeNode1, treeNode2);
treeNode1.setChildren(treeNode3, treeNode4);
treeNode2.setChildren(treeNode5, treeNode6);

return treeNode0;
}
}