103. Binary Tree Zigzag Level Order Traversal #29
Conversation
| node = nodes.pop() | ||
| if from_right: | ||
| zigzag_level_ordered_values[-1].append(node.val) | ||
| append_node_if_exist(node.right) |
There was a problem hiding this comment.
一見して何にノードを入れるのかわからないと思いました。
ノードを入れる対象のオブジェクトも引数にしていい気がします。
There was a problem hiding this comment.
はい、どうかなと思ったんですが、やはり突っ込まれましたね。
| zigzag_level_ordered_values[level] \ | ||
| = zigzag_level_ordered_values[level] + [node.val] |
There was a problem hiding this comment.
イコールの前で改行する＆イコールの位置が前の行の頭と同じなのあまり見ない気がしました。書くならこっち↓の方が自然かも？
zigzag_level_ordered_values[level] = \
zigzag_level_ordered_values[level] + [node.val]あとelse側と書き方を揃えたのかなと思いつつ、無駄にコピーが走るのでこちらは
zigzag_level_ordered_values[level].append(node.val)でいい気がしました。
| return [] | ||
|
|
||
| zigzag_level_ordered_values = [] | ||
| def zigzag_level_order_helper(nodes: list[Optional[TreeNode]], level: int) -> None: |
There was a problem hiding this comment.
nodesの型、 list[TreeNode] じゃないですか？nodesの要素にNoneが入るケースは無い気がします
There was a problem hiding this comment.
そうですね。
深く考えずコピペしてしまいました。。
| num_same_level_nodes = len(nodes) | ||
| for _ in range(num_same_level_nodes): |
There was a problem hiding this comment.
num_same_level_nodesは作る必要ない気がします。 for _ in range(len(nodes)) で良いと思いました。さらに、 for node in reversed(nodes) でpopしないか while nodes でpopするで良い気がしました。
| zigzag_level_ordered_values[level] \ | ||
| = [node.val] + zigzag_level_ordered_values[level] |
There was a problem hiding this comment.
zigzag_level_ordered_values[level].insert(0, node.val) の方がコピーを生成しない分効率が良いと思います。
| while level >= len(zigzag_level_ordered_values): | ||
| zigzag_level_ordered_values.append([]) |
There was a problem hiding this comment.
レベル順に見ていっているので zigzag_level_ordered_values.append([]) だけでも個人的にはパズル感はありませんが、どうでしょうか。
| if from_right: | ||
| values.reverse() | ||
|
|
||
| zigzag_level_ordered_values.append(values) |
There was a problem hiding this comment.
こちらのように、レベルごとに値を集めて最後に答えに追加する書き方の方が読みやすかったです。
https://leetcode.com/problems/binary-tree-zigzag-level-order-traversal/description/