543. Diameter of Binary Tree

[naoto-iwase]

543. Diameter of Binary Tree

• URL: https://leetcode.com/problems/diameter-of-binary-tree/
• Difficulty: Easy
• Tags: Tree, Depth-First Search, Binary Tree
• Notebook: https://share.solve.it.com/d/97edf75b8b065147544543f0b244941f

---

Next: 57. Insert Interval

[nodchip]

• 行きがけは left_output と right_output は空
• output が親ノードの left_output または right_output を指す
• 帰りがけに left_output と right_output が非空になっている

ということですね。普段あまり目にしない書き方だったため、読むのにやや時間がかかりました。
チームの平均的なソフトウェアエンジニアが現実的な時間で読めるのであれば、この書き方でも良いと思います。そうでない場合は、再帰で書いたほうが良いかもしれません。

[naoto-iwase]

ありがとうございます。ご理解の通りです。

リストの空/非空を「仕事完了フラグ」として兼用しているのが分かりにくさの原因だと思います。以下のようにdataclassで明示的にすれば多少改善しそうですが、それでもrecursiveの方が素直ですね。今回は練習としてiterativeで書いてみました。

from dataclasses import dataclass

@dataclass
class Job:
    done: bool = False
    diameter: Optional[int] = None
    depth: Optional[int] = None

result = Job()
...