102. Binary Tree Level Order Traversal

[fhiyo]

https://leetcode.com/problems/binary-tree-level-order-traversal/description/

[sakupan102]

個人的にはappend_if_existsで左右のノードを調べるほうが良いかと思いました。

[fhiyo]

class Solution:
    def levelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
        def append_if_child_exists(nodes: list[TreeNode], node: TreeNode):
            if node.left:
                nodes.append(node.left)
            if node.right:
                nodes.append(node.right)

        if not root:
            return []
        nodes = [root]
        level_order_values = []
        while nodes:
            values = []
            next_level_nodes = []
            for node in nodes:
                values.append(node.val)
                append_if_child_exists(next_level_nodes, node)
            nodes = next_level_nodes
            level_order_values.append(values)
        return level_order_values

関数名はもう少しあるかもですが、こんな感じですかね？発想なかったんですがなるほどと思いました。スッキリすると思います

[Mike0121]

currentに関して、自分のPRで以前コメントをいただいたのでリンク貼っておきます。
Mike0121/LeetCode#7

[Mike0121]

valuesだけだと、同じ深さのnodeの値をまとめた配列、という内容には少し物足りない気がしました。level_valueとかでしょうか？

[fhiyo]

配列なので level_values ですかね？自分もvaluesは微妙だなと思いつつ、ギリギリ分かるかなと思ってこうしてみてました。
level_valuesだとlevel_order_valuesとの棲み分けが分かりにくくなるかも？とは思いましたが...うーんどうなんですかね (level_order_valuesという名前が微妙なのかもしれない)

[hayashi-ay]

良いと思います。