题目:输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
这题可以用递归做,较为简单,代码如下:
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
class Solution {
public:
int TreeDepth(TreeNode* pRoot)
{
if(pRoot==NULL)
return 0;
int leftNodeDepth=TreeDepth(pRoot->left);
int rightNodeDepth=TreeDepth(pRoot->right);
return max(leftNodeDepth,rightNodeDepth)+1;
}
};
随后,该题目又出了一个扩展题目,让判断是否是一棵平衡二叉树。平衡二叉树就是任意结点的左右子树深度相差不超过1,代码如下:
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
if(pRoot==NULL)
return true;
int leftLength=TreeDepth(pRoot->left);
int rightLength=TreeDepth(pRoot->right);
if(leftLength-rightLength>1||rightLength-leftLength>1)
return false;
else
return IsBalanced_Solution(pRoot->left)&&IsBalanced_Solution(pRoot->right);
}
int TreeDepth(TreeNode* pRoot)
{
if(pRoot==NULL)
return 0;
int leftNodeDepth=TreeDepth(pRoot->left);
int rightNodeDepth=TreeDepth(pRoot->right);
return max(leftNodeDepth,rightNodeDepth)+1;
}
};