看题解上没有介绍使用层次遍历计算二叉树的深度的思路,那我就介绍一下吧。
思路:

  • 借助队列,对二叉树进行层次遍历;
  • 在层次遍历的过程中,每次当队列中某一层的节点出队完成后,高度+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;
      }