104. Maximum Depth of Binary Tree

[fhiyo]

https://leetcode.com/problems/maximum-depth-of-binary-tree/description/

[liquo-rice]

nodes = []
append_if_exists(node, nodes)

[liquo-rice]

optional: 上のように書いてもいいですね。

[liquo-rice]

引数の順番は、コンテナ、アイテムの順番が一般的かもしれません。（上ではそうなってますね）
https://docs.python.org/3/library/heapq.html#heapq.heappush

[liquo-rice]

これの再帰版もありますね。

[fhiyo]

なるべく再現するとこんな感じですかね。

class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if root is None:
            return 0
        max_depth = 0

        def max_depth_helper(node: Optional[TreeNode], depth: int):
            nonlocal max_depth
            max_depth = max(max_depth, depth)
            if node.left is not None:
                max_depth_helper(node.left, depth + 1)
            if node.right is not None:
                max_depth_helper(node.right, depth + 1)

        max_depth_helper(root, 1)
        return max_depth

少し整理してみたらこうなりました。

class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        max_depth = 0

        def max_depth_helper(node: Optional[TreeNode], depth: int):
            nonlocal max_depth
            max_depth = max(max_depth, depth)
            if node is None:
                return
            max_depth_helper(node.left, depth + 1)
            max_depth_helper(node.right, depth + 1)

        max_depth_helper(root, 0)
        return max_depth

[oda]

一応、これなくても動きますかね。まあ、書いておいてもいいですよね。

[fhiyo]

や、問題設定上 The number of nodes in the tree is in the range [0, 10^4]. なので、入力でrootがNoneな場合があり得ますね (そして node.left のところでエラーになる)

[TORUS0818]

個人的にやや複雑な処理だなと感じたんですが、割とサラッと実装できる感じでしょうか。

[fhiyo]

あまり覚えてないんですが、そこまでさらっとは出来なかった気がします...

[nodchip]

個人的には deque に (node, depth) を入れ、先頭から取り出し、末尾に入れていくタイプの実装のほうが好きですが、fhiyo さんの実装でもよいと思います。