【面试题39】二叉树的深度

【题目描述】

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

【解决方案】

递归解决。

我的代码实现，仅供参考：

        public static int TreeDepth(TreeNode root)
        {
            if (root == null)
                return 0;
            
            int leftDepth = TreeDepth(root.left);
            int rightDepth = TreeDepth(root.right);

            return leftDepth > rightDepth ? (leftDepth + 1) : (rightDepth + 1);
        }

【本题扩展】

输入一棵二叉树的根结点，判断该树是不是平衡二叉树。

解法一：需要重复遍历结点多次的解法，简单但不足以打动面试官

我的代码实现，仅供参考：

        public static bool IsBalanced(TreeNode root)
        {
            if (root == null)
                return true;

            int leftDepth = TreeDepth(root.left);
            int rightDepth = TreeDepth(root.right);

            if (Math.Abs(leftDepth - rightDepth) > 1)
                return false;

            return IsBalanced(root.left) && IsBalanced(root.right);
        }

        public static int TreeDepth(TreeNode root)
        {
            if (root == null)
                return 0;
            
            int leftDepth = TreeDepth(root.left);
            int rightDepth = TreeDepth(root.right);

            return leftDepth > rightDepth ? (leftDepth + 1) : (rightDepth + 1);
        }

解法二：每个结点只遍历一次，正式面试官喜欢的

我的代码实现，仅供参考：

        public static bool IsBalanced(BinaryTreeNode root)
        {
            int depth = 0;

            return IsBalanced(root, ref depth);
        }

        public static bool IsBalanced(BinaryTreeNode root, ref int depth)
        {
            if (root == null)
                return true;

            int left = 0, right = 0;

            if (IsBalanced(root.Left, ref left) && IsBalanced(root.Right, ref right))
            {
                if (Math.Abs(left - right) <= 1)
                {
                    depth = 1 + (left>right? left :right);
                    return true;
                }
            }

            return false;
        }