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==NULL) return 0;
        
int l=maxDepth(root->left)+1; int r=maxDepth(root->right)+1; return l>r?l:r; } }; //深度優先:用棧的迴圈版 class Solution { public: int maxDepth(TreeNode* root) { if(root==NULL) return 0; stack<pair<TreeNode*,int>> s; TreeNode* p=root;
int Maxdeep=0; int deep=0; while(!s.empty()||p!=NULL)//若棧非空,則說明還有一些節點的右子樹尚未探索;若p非空,意味著還有一些節點的左子樹尚未探索 { while(p!=NULL)//優先往左邊走 { s.push(pair<TreeNode*,int>(p,++deep)); p=p->left; } p=s.top().first;//
若左邊無路,就預備右拐。右拐之前,記錄右拐點的基本資訊 deep=s.top().second; if(Maxdeep<deep) Maxdeep=deep;//預備右拐時,比較當前節點深度和之前儲存的最大深度 s.pop();//將右拐點出棧;此時棧頂為右拐點的前一個結點。在右拐點的右子樹全被遍歷完後,會預備在這個節點右拐 p=p->right; } return Maxdeep; } }; //廣度優先:使用佇列 class Solution { public: int maxDepth(TreeNode* root) { if(root==NULL) return 0; deque<TreeNode*> q; q.push_back(root); int deep=0; while(!q.empty()) { deep++; int num=q.size(); for(int i=1;i<=num;i++) { TreeNode* p=q.front(); q.pop_front(); if(p->left) q.push_back(p->left); if(p->right) q.push_back(p->right); } } return deep; } };