LeetCode-104.二叉树的最大深度

原创于 2019-11-26 21:02:18 发布 · 140 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一种使用深度优先遍历(DFS)算法来寻找二叉树最大深度的方法。通过设置全局变量保存最大深度，从根节点开始遍历，遇到叶子节点时检查并更新最大深度。代码实现简洁，易于理解。

解析
设置全局变量保存最大深度，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 max_depth = -1;
    void dfs(TreeNode *t, int cnt){
        if(!t->left && !t->right){
            max_depth = cnt > max_depth ? cnt : max_depth;
            return;
        }
        if(t->left) dfs(t->left, cnt + 1);
        if(t->right) dfs(t->right, cnt + 1);
    }
    int maxDepth(TreeNode* root) {
        if(!root) return 0;
        dfs(root, 1);
        return max_depth;
    }
};