104. Maximum Depth of Binary Tree

[ryosuketc]

104. Maximum Depth of Binary Tree

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

[Fuminiton]

nodes_and_depthsなどとしてもいい気がします

[Fuminiton]

特に問題ないと思います。
再帰への書き換え、queueを使う/使わないBFS等の実装もありました

[oda]

先に進んでもいいかもしれません。一応、こちらに(別の問題ですが) Python のバージョンがあるので、これをみたらやろうとしていること分かるかもしれません。
potrue/leetcode#22 (comment)

[ryosuketc]

一応機械的に namedtuple で書き直してみましたが、結局あまりよくわかりませんでした。一旦先に進んで、どこかで戻ってみようかと思います… (return_value, left_value, right_value が何なのかすらわかっていない)。

class Solution:
    def minDepth(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0
            
        StackEntry = namedtuple('StackEntry', ['node', 'depth', 'return_value', 'left_value', 'right_value'])
        result = []
        stack = [StackEntry(root, 1, result, [], [])]
        while stack:
            node, depth, return_value, left_value, right_value = stack[-1]
            if node.left is not None and not left_value:
                stack.append(StackEntry(node.left, depth + 1, left_value, [], []))
                continue
            if node.right is not None and not right_value:
                stack.append(StackEntry(node.right, depth + 1, right_value, [], []))
                continue
            if not left_value and not right_value:
                min_depth = depth
            else:
                min_depth = min(value[0] for value in (left_value, right_value) if value)
            return_value.append(min_depth)
            stack.pop()
        return result[0]

[ryosuketc]

ここでの議論が参考になりそうなので記録しておきます。

https://github.com/ryosuketc/leetcode_arai60/pull/23/files#r2122113958