用一個隊列來保存將要打印的結(jié)點(diǎn)。為了把二叉樹的每一行單獨(dú)打印到一行里,我們需要兩個變量:一個變量表示在當(dāng)前的層中還沒有打印的結(jié)點(diǎn)數(shù),另一個變量表示下一次結(jié)點(diǎn)的數(shù)目。
private static class BinaryTreeNode {
private int val;
private BinaryTreeNode left;
private BinaryTreeNode right;
public BinaryTreeNode() {
}
public BinaryTreeNode(int val) {
this.val = val;
}
@Override
public String toString() {
return val + "";
}
}
import java.util.LinkedList;
import java.util.List;
public class Test60 {
private static class BinaryTreeNode {
private int val;
private BinaryTreeNode left;
private BinaryTreeNode right;
public BinaryTreeNode() {
}
public BinaryTreeNode(int val) {
this.val = val;
}
@Override
public String toString() {
return val + "";
}
}
/**
* 題目:從上到下按層打印二叉樹,同一層的結(jié)點(diǎn)按從左到右的順序打印,每一層打印一行。
* @param root
*/
public static void print(BinaryTreeNode root) {
if (root == null) {
return;
}
List<BinaryTreeNode> list = new LinkedList<>();
BinaryTreeNode node;
// 當(dāng)前層的結(jié)點(diǎn)個數(shù)
int current = 1;
// 記錄下一層的結(jié)點(diǎn)個數(shù)
int next = 0;
list.add(root);
while (list.size() > 0) {
node = list.remove(0);
current--;
System.out.printf("%-3d", node.val);
if (node.left != null) {
list.add(node.left);
next++;
}
if (node.right != null) {
list.add(node.right);
next++;
}
if (current ==0) {
System.out.println();
current = next;
next = 0;
}
}
}
public static void main(String[] args) {
BinaryTreeNode n1 = new BinaryTreeNode(1);
BinaryTreeNode n2 = new BinaryTreeNode(2);
BinaryTreeNode n3 = new BinaryTreeNode(3);
BinaryTreeNode n4 = new BinaryTreeNode(4);
BinaryTreeNode n5 = new BinaryTreeNode(5);
BinaryTreeNode n6 = new BinaryTreeNode(6);
BinaryTreeNode n7 = new BinaryTreeNode(7);
BinaryTreeNode n8 = new BinaryTreeNode(8);
BinaryTreeNode n9 = new BinaryTreeNode(9);
n1.left = n2;
n1.right = n3;
n2.left = n4;
n2.right = n5;
n3.left = n6;
n3.right = n7;
n4.left = n8;
n4.right = n9;
print(n1);
}
}
http://wiki.jikexueyuan.com/project/for-offer/images/78.png" alt="" />