104. 二叉树的最大深度

题目描述

给定一个二叉树，找出其最大深度。二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

说明: 叶子节点是指没有子节点的节点。

 

示例

给定二叉树 [3,9,20,null,null,15,7]，

    3
   / \
  9  20
    /  \
   15   7
返回它的最大深度 3 。

 

思路1

dfs，只要比较左子树和右子树中最高值，然后加上根节点。

 

代码

/**
 * 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;
        int l = maxDepth(root->left);
        int r = maxDepth(root->right);
        return max(l, r) + 1;
    }
};

 

思路2

bfs：利用一个栈来实现，同时记录当前结点高度。

 

代码

    int maxDepth(TreeNode* root) {
        if (!root) return 0;
        stack<pair<TreeNode*, int>> s;
        int maxDepth = 0, curDepth = 0;
        while (!s.empty() || root){
            while (root){
                s.emplace(pair<TreeNode*, int>(root, ++curDepth));
                maxDepth = max(maxDepth, curDepth);
                root = root->left;
            }
            if (!s.empty()){
                TreeNode* cur = s.top().first;
                curDepth = s.top().second;
                s.pop();
                if (cur->right) root = cur->right;
            }
        }
        return maxDepth;
    }