剑指offer——二叉树的深度（虽然都是递归，但递归也有不同的写法，or循环）

题目描述
输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。

思路：
给的入口是一个二叉树的根节点，所以应该是深度优先遍历，然后不断更新最大深度。（？）

掌握递归和普通循环的写法。

---

递归：

public class Solution {
    int max = 0;
    public int TreeDepth(TreeNode root) {
        if(root==null)
            return 0;
        getDepth(root,0);
        return max;
    }

    public void getDepth(TreeNode root, int now){
        if(root==null){
            if(now>max)
                max = now;
            return; // 一开始把return忘了
        }
        now++;
        getDepth(root.left,now);
        getDepth(root.right,now);
    }
}

换一种递归写法：

public class Solution {
    public int TreeDepth(TreeNode root) {
        if(root==null)
            return 0;
        return getDepth(root,0,0);
    }

    public int getDepth(TreeNode root, int now, int max){
        if(root==null){
            if(now>max)
                max = now;
            return max;
        }
        now++;
        max = getDepth(root.left,now,max);
        max = getDepth(root.right,now,max);
        return max;
    }
}

别人的递归

思路：该树的深度等于其左右子树的最大深度+1。

public class Solution {
    public int TreeDepth(TreeNode pRoot)
    {
        if(pRoot == null){
            return 0;
        }
        int left = TreeDepth(pRoot.left);
        int right = TreeDepth(pRoot.right);
        return Math.max(left, right) + 1;
    }
}

非递归

import java.util.Queue;
import java.util.LinkedList;

public class Solution {
    public int TreeDepth(TreeNode pRoot)
    {
        if(pRoot == null){
            return 0;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(pRoot);
        int depth = 0, count = 0, nextCount = 1;
        while(queue.size()!=0){
            TreeNode top = queue.poll();
            count++;
            if(top.left != null){
                queue.add(top.left);
            }
            if(top.right != null){
                queue.add(top.right);
            }
            if(count == nextCount){
                nextCount = queue.size();
                count = 0;
                depth++;
            }
        }
        return depth;
    }
}

depth：当前节点所在的层数，count已经遍历了的节点数，nextCount下层的节点总数；当count==nextCount的时候，代表本层的节点已经遍历完毕。

或者只维护一个count值，表示上一层的节点数，然后在while循环里用一个for循环遍历完，更直观一点