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指向棧頂結點的右子結點