剑指offer（三十八）——二叉树的深度

剑指offer（三十八）——二叉树的深度

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

初始版

讲道理，这代码写的有点烂，定义了全局变量还递归写的也不咋地。我对递归也就这点认识（惭愧）。

	public int tree_deepth;
	
	public int TreeDepth(TreeNode root) {
        tree_deepth = 0;
        DFS(root, 0);
        return tree_deepth;
    }
	
	public void DFS(TreeNode root, int deepth) {
		if (root == null) {
			return;
		}
		deepth ++;
		if (tree_deepth < deepth) {
			tree_deepth = deepth;
		}
		if (root.right != null) {
			DFS(root.right, deepth);
		}
		if (root.left != null) {
			DFS(root.left, deepth);
		}
	}

优化一

• 本函数递归+不用全局变量
• 参考大佬写法进行优化
• 直接到达叶子节点进行返回，每一次返回深度加一，取最深的即可。图示各层返回情况：
  

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

优化二

• 非递归+不用全局变量
• 其实就是层序遍历了：

	public int TreeDepth(TreeNode root) {
        if (root == null) {
			return 0;
		}
        int depth = 0;
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        while(!queue.isEmpty()) {
        	depth++;
        	//清除每一层的结点
        	int size = queue.size();
        	for(int i = 0; i < size; i++) {
        		TreeNode temp = queue.poll();
        		if (temp.right != null) {
					queue.add(temp.right);
				}
        		if (temp.left != null) {
					queue.add(temp.left);
				}
        	}
        }
        return depth;
    }