103. Binary Tree Zigzag Level Order Traversal

[fhiyo]

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

[YukiMichishita]

Suggested change

	level_order_values.append(values)
	zigzag_level_order_helper(next_level_nodes, not from_left)
	zigzag_level_order_helper([root], True)
	return level_order_values
	yield values
	yield from zigzag_level_order_helper(next_level_nodes, not from_left)
	return list(zigzag_level_order_helper([root], True))

yieldを使うと副作用に頼らなくてよいので、流儀によってはこちらの方が好まれるかもしれないと思いました。

[fhiyo]

ありがとうございます。こちらの書き方の方がhelperからそのまま値を返せるので良さそうに感じました

[Mike0121]

全体的に良いと思いました。valuesという変数名、102. Binary Tree Level Order Traversalと同様もう少し説明があっても良いかと思いました。

[sakupan102]

良いと思いました！
恐らく好みの問題ですがfor ループで後ろから見ていってもよい気がしました。

[oda]

これが一番読みやすい気はしますよね。意図がはっきりしています。

実務では速度よりも意図の伝わりやすさを取ることが多いです。

[TORUS0818]

これDFSになってますか？

[fhiyo]

なってないですねこれ。ありがとうございます。
辞書の挿入順が保存されるPython3.7以降の環境を前提に書くならこんな感じですかね。

class Solution:
    def zigzagLevelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
        if not root:
            return []
        level_to_values = defaultdict(deque)
        stack = [(root, 0)]
        while stack:
            node, level = stack.pop()
            if level % 2 == 0:
                level_to_values[level].append(node.val)
            else:
                level_to_values[level].appendleft(node.val)
            # To visit the left node first, we push the right node to the stack first.
            if node.right:
                stack.append((node.right, level + 1))
            if node.left:
                stack.append((node.left, level + 1))
        # We don't need to sort level_to_values.items(),
        # because the key-value pairs of level_to_values must be added in ascending order by level.        
        return [list(values) for values in level_to_values.values()]

[TORUS0818]

実装ありがとうございました！

[nodchip]

left_to_right という変数名でも良いと思います。