看题解上没有介绍使用层次遍历计算二叉树的深度的思路,那我就介绍一下吧。
思路:
- 借助队列,对二叉树进行层次遍历;
- 在层次遍历的过程中,每次当队列中某一层的节点出队完成后,高度+1;
- 关键点:判别队列中某一层节点出队完成的标准是什么?
在出队之前,此时队列中记录的只有某一层节点,所以队列的大小就是某一层节点的个数。当此个数减到0的时候,则说明该层节点全部出队完成public int TreeDepth2(TreeNode root) { if(root == null) return 0; Queue<TreeNode> queue = new LinkedList(); queue.add(root); int high = 0; int size;//用size控制high增长的次数和时机(同一层的元素没有完全退出队列的时候high不可以增加) TreeNode node; while(queue.size() != 0){ size = queue.size();//队列长度 while(size != 0){ node = queue.poll(); if(node.left != null) queue.add(node.left); if(node.right != null) queue.add(node.right); size--;//当size==0时说明同一层的元素遍历完成 } high++; } return high; }