144. 二叉樹的前序遍歷

知識點:二叉樹;遞迴;Morris遍歷

題目描述

給你二叉樹的根節點 root ,返回它節點值的 前序 遍歷。

示例

輸入:root = [1,null,2,3]
輸出:[1,2,3] 輸入:root = []
輸出:[] 輸入:root = [1]
輸出:[1] 輸入:root = [1,2]
輸出:[1,2] 輸入:root = [1,null,2]
輸出:[1,2]

解法一:遞迴

/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
List<Integer> list = new ArrayList<>();
public List<Integer> preorderTraversal(TreeNode root) {
if(root == null) return list;
list.add(root.val);
preorderTraversal(root.left);
preorderTraversal(root.right);
return list;
}
}

時間複雜度;0(N),每個節點恰好被遍歷一次;

空間複雜度;O(N),遞迴過程中棧的開銷;

解法二:迭代法

入棧一定是先右後左,這樣出來才能使先左後右;

壓入根節點;

1.彈出就列印;

2.如有右孩子,壓入右;

3.如有左孩子,壓入左;

/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
List<Integer> list = new ArrayList<>();
if(root != null) stack.push(root);
while(!stack.isEmpty()){
TreeNode top = stack.pop();
list.add(top.val);
if(top.right != null) stack.push(top.right);
if(top.left != null) stack.push(top.left);
}
return list;
}
}

解法三:Morris遍歷

構建從下到上的連線,一條路能夠走遍所有節點;

/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
List<Integer> list = new ArrayList<>();
public List<Integer> preorderTraversal(TreeNode root) {
if(root == null) return list;
TreeNode cur = root;
TreeNode mostRightNode = null;
while(cur != null){
mostRightNode = cur.left;
if(mostRightNode != null){
//有左子樹;
while(mostRightNode.right != null && mostRightNode.right != cur){
mostRightNode = mostRightNode.right; //找到左子樹的最右節點;
}
if(mostRightNode.right == null){
mostRightNode.right = cur; //構建向上的連線;
list.add(cur.val);
cur = cur.left;
continue;
}else{
//第二次到節點,斷開連線
mostRightNode.right = null;
cur = cur.right;
}
}else{
list.add(cur.val);
cur = cur.right;
}
}
return list;
}
}

體會

二叉樹的遍歷是二叉樹的最基礎的,不僅要掌握遞迴寫法,迭代法和morris遍歷也需要掌握,更多詳細可參考這篇總結二叉樹

相關題目

94. 二叉樹的中序遍歷

145. 二叉樹的後序遍歷

相關連結

二叉樹