题目及测试
package sword023;
import java.util.List;
/*从上往下打印出二叉树的每个节点,同层节点从左至右打印。
*/
public class main {
public static void main(String[] args) {
Object[] x=new Object[]{1,2,2,3,4,4,3};
BinaryTree tree=new BinaryTree(x);
BinaryTree.printTree(tree.root);
test(tree.root);
Object[] x2=new Object[]{3,9,20,null,null,15,7};
BinaryTree tree2=new BinaryTree(x2);
BinaryTree.printTree(tree2.root);
test(tree2.root);
}
private static void test(TreeNode ito) {
Solution solution = new Solution();
List<Integer> rtn;
long begin = System.currentTimeMillis();
rtn = solution.levelOrder(ito);//执行程序
long end = System.currentTimeMillis();
System.out.println("rtn=" );
for (Integer integer : rtn) {
System.out.print(integer+" ");
}
System.out.println();
System.out.println("耗时:" + (end - begin) + "ms");
System.out.println("-------------------");
}
}
解法1(成功)
按照层次遍历的方法,使用队列辅助。
1.将根结点加入队列。
2.循环出队,打印当前元素,若该结点有左子树,则将其加入队列,若有右子树,将其加入队列。
3.直到队列为空,表明已经打印完所有结点。
package sword023;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Integer> levelOrder(TreeNode root) {
List<Integer> result=new ArrayList<>();
if(root==null){
return result;
}
Queue<TreeNode> queue=new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()) {
TreeNode now = queue.poll();
result.add(now.val);
if(now.left != null) {
queue.add(now.left);
}
if(now.right != null) {
queue.add(now.right);
}
}
return result;
}
}