每天一算:Binary Tree Preorder Traversal

LeetCode上第144 號問題:二叉樹的前序遍歷

題目

給定一個二叉樹,返回它的 前序 遍歷。 示例: 輸入: [1,null,2,3] 輸出: [1,2,3] 進階: 遞歸算法很簡單,你可以通過迭代算法完成嗎?

解題思路

棧(Stack)的思路來處理問題。

前序遍歷的順序爲根-左-右,具體算法爲:

  • 把根節點push到棧中
  • 循環檢測棧是否爲空,若不空,則取出棧頂元素,保存其值
  • 看其右子節點是否存在,若存在則push到棧中
  • 看其左子節點,若存在,則push到棧中。

動畫演示

動畫演示GIF加載有點慢,請稍待片刻加載顯示^_^

參考代碼

補充

下面這種寫法使用了一個輔助結點p,這種寫法其實可以看作是一個模版,對應的還有中序和後序的模版寫法,形式很統一,方便於記憶。後續更新的中序和後序文章中都會補充該寫法。思路與代碼如下:

  • 輔助結點p初始化爲根結點,while循環的條件是棧不爲空或者輔助結點p不爲空
  • 在循環中首先判斷如果輔助結點p存在,那麼先將p加入棧中,然後將p的結點值加入結果res中,此時p指向其左子結點
  • 否則如果p不存在的話,表明沒有左子結點,我們取出棧頂結點,將p指向棧頂結點的右子結點
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章