【LeetCode】104. Maximum Depth of Binary Tree(DFS|BFS)

원제


Given a binary tree, find its maximum depth.
The maximum depth is the number of nodes along the longest path from the root node down to the farthest leaf node.

분석


간단한 기본 문제는 BFS, DFS를 사용하여 수행할 수 있습니다.
주로 비귀속 방식의 편찬을 연습할 수 있다.

나의 해답


C++ BFS 버전:
/** * 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 bfs(struct TreeNode* root) {
        int ans = 1;
        queue< pair<TreeNode*,int> > q;
        q.push({root, 1});
        while (!q.empty()) {
            TreeNode *cur = q.front().first;
            int dep = q.front().second;
            q.pop();
            if (dep > ans)  ans = dep;
            if (cur->left)  q.push({cur->left, dep+1});
            if (cur->right) q.push({cur->right, dep+1});
        }
        return ans;
    }

    int maxDepth(TreeNode* root) {
        if (root)
            return bfs(root);
        return 0;
    }
};

C++ DFS 비귀속 버전:
class Solution {
public:


    int dfs(struct TreeNode* root) {
        int ans = 1;
        stack< pair<TreeNode*,int> > s;
        s.push({root, 1});
        while (!s.empty()) {
            TreeNode *cur = s.top().first;
            int dep = s.top().second;
            s.pop();
            if (dep > ans)  ans = dep;
            if (cur->left)  s.push({cur->left, dep+1});
            if (cur->right) s.push({cur->right, dep+1});
        }
        return ans;
    }

    int maxDepth(TreeNode* root) {
        if (root)
            return dfs(root);
        return 0;
    }
};

하하, 사실 DFS와 BFS의 코드는 기본적으로 같다. 단지 하나의 사용 대기열, 하나의 사용 창고일 뿐이다.

네티즌의 답변


C++ DFS 재귀속:
class Solution {
public:
    int maxDepth(TreeNode* root) {
        if (root) return 1+ max(maxDepth(root->left), maxDepth(root->right));
        return 0;
    }
};

Python:
def maxDepth(self, root):
    return 1 + max(map(self.maxDepth, (root.left, root.right))) if root else 0

아래와 같은 가격:
if root == None:
        return 0
return max(self.maxDepth(root.left),self.maxDepth(root.right)) + 1

좋은 웹페이지 즐겨찾기