求二叉樹深度是一個基本的算法,解決的思路如下:
1. 根據二叉樹的非遞歸遍歷,設置兩個記錄變量“觀察”棧的深度變化:real_depth是棧真正的深度,cur_depth是當前棧的深度。對每個節點進行遍歷,如果cur_depth大於real_depth則更新real_depth。
2. 利用遞歸求解。樹的深度爲其左右子樹中較大者的深度加1。思想很好理解,代碼也簡單。
以下代碼爲遞歸求解:
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int maxDepth(TreeNode* root) {
if (!root)
return 0;
return 1 + max(maxDepth(root->left), maxDepth(root->right));
}
};