題目描述
輸入一棵二叉樹,求該樹的深度。從根結點到葉結點依次經過的結點(含根、葉結點)形成樹的一條路徑,最長路徑的長度爲樹的深度。
題解
#include <iostream>
#include <algorithm>
using namespace std;
struct TreeNode
{
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
TreeNode() {}
};
TreeNode* newTree()
{
TreeNode* node = new TreeNode;
int x;
cin >> x;
if (!x)node = NULL;
else
{
node->val = x;
node->left = newTree();
node->right = newTree();
}
return node;
}
int TreeDepth(TreeNode* root)
{
if (!root)return 0;
return max(TreeDepth(root->left) + 1, TreeDepth(root->right) + 1);
}
int main()
{
ios::sync_with_stdio(false);
TreeNode* root = NULL;
root = newTree();
cout << TreeDepth(root) << endl;
return 0;
}