NOWCODER【程式設計題】——二叉樹的深度
阿新 • • 發佈:2019-02-15
題目描述
輸入一棵二叉樹,求該樹的深度。從根結點到葉結點依次經過的結點(含根、葉結點)形成樹的一條路徑,最長路徑的長度為樹的深度。
程式碼實現:
二叉樹節點定義:
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
方法一 遞迴:二叉樹為空時,返回0;否則,二叉樹的深度(高度)等於二叉樹左子樹高度和右子樹高度中較大值加上1(加上一個根節點),然後左子樹的高度又等於左子樹的左子樹高度和左子樹的右子樹的高度中較大值加1....這是一個不斷求解子問題的過程,也就是遞迴。
class Solution {
public:
int TreeDepth(TreeNode* pRoot)
{
if(pRoot==NULL)return 0;
int Dleft=TreeDepth(pRoot->left);
int Dright=TreeDepth(pRoot->right);
int depth=Dleft>Dright?Dleft+1:Dright+1;
return depth;
}
};
方法二 非遞迴:建立一個佇列用來進行層序遍歷二叉樹,每次帶入一層節點進去佇列中,記錄二叉樹的層數(即深度),具體看以下程式碼。
class Solution { public: int TreeDepth(TreeNode* pRoot) { if(pRoot==NULL)return 0; queue<TreeNode*>q; q.push(pRoot); int depth=0; while(!q.empty()) { ++depth; for(int i=0;i<q.size();++i) { TreeNode*temp=q.front(); q.pop(); if(temp->left) q.push(temp->left); if(temp->right) q.push(temp->right); } } return depth; } };