104 Maximum Depth of Binary Tree

[MA-yo-TA]

この問題：https://leetcode.com/problems/maximum-depth-of-binary-tree/
次の問題：https://leetcode.com/problems/minimum-depth-of-binary-tree/

[h-masder]

next_layerの初期化はまとめられそうです。

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

        nodes = [root]
        depth = 0
        while nodes:
            depth += 1
            next_layer = []
            for node in nodes:
                if node.left is not None:
                    next_layer.append(node.left)
                if node.right is not None:
                    next_layer.append(node.right)
            nodes = next_layer
            
        return depth

[MA-yo-TA]

ありがとうございます。
個人的には nodes と next_layer の初期化・更新はセットでやりたい感覚（今の階層 vs. 次の階層、という対の意識）があってこう書いたのですが h-masder さんの感覚だとどうですか？

nodes はループ中も一貫した意味を持つ名前なのに対して next_layer は while のループ一回ごとに完結して意味を持つのでループ内だけに存在してて欲しい（からまとめる）というのもわかります。

[h-masder]

私としては、next_layer はループの中で毎回作るものというイメージがあるので、中で初期化しています。

とはいえ、このあたりは好みの問題で、どちらの書き方でもいいのかなと思っています。

感覚としては、「各ループの最初に next_layer を空にして使い始める」か、「ループの最後に次のループのために空にしておく（後片付けしておく）」かという違いで、どちらの考え方もよくわかります。

[MA-yo-TA]

ありがとうございます。次の問題（https://github.com/MA-yo-TA/leetcode/pull/22）含めて色々書いてみた結果、ループの最後に後片付けするのが自分としてはしっくりくると感じることができました。

[Shunii85]

いいと思いました。