求树的深度

题目描述

　　输入一棵二叉树，求该树的深度。

　　从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。

来自牛客。

    public int TreeDepth(TreeNode root) {
        //递归实现
        /*if(root == null){
            return 0;
        }
        int leftcount = TreeDepth(root.left);
        int rightcount = TreeDepth(root.right);
        //如果左右子节点都为空，返回0,0再加当前节点1
        return Math.max(leftcount , rightcount)+1;
        */
        if(root == null) return 0;
        LinkedList<TreeNode> list = new LinkedList<TreeNode>();
        list.add(root);
        //depth为深度，什么情况下深度会加1呢？当把当前这一层所有节点都pop完就进入下一层
        //那怎么确定计算完当前节点呢？cur累加至当层左右节点pop完，即cur==nextcount
        int depth = 0,curCount = 0,nextCount = list.size();
        while(!list.isEmpty()){
            TreeNode curTree = list.pop();
            curCount++;
            if(curTree.left != null){
                list.add(curTree.left);
            }
            if(curTree.right != null){
                list.add(curTree.right);
            }
            if(curCount == nextCount){
                curCount = 0;
                nextCount = list.size();
                depth++;
            }
        }
        return depth;


    }