144. 二叉树的前序遍历

给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

示例 1:

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

示例 2:

1
2
输入:root = []
输出:[]

示例 3:

1
2
输入:root = [1]
输出:[1]

示例 4:

1
2
输入:root = [1,2]
输出:[1,2]

示例 5:

1
2
输入:root = [1,null,2]
输出:[1,2]

提示:

  • 树中节点数目在范围 [0, 100]
  • -100 <= Node.val <= 100

进阶:递归算法很简单,你可以通过迭代算法完成吗?

题解:

递归法:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
public List<Integer> preorderTraversal(TreeNode root) {
//list用于保存前序遍历的节点序列
List<Integer> list = new ArrayList<>();
preOrder(list,root);
return list;
}
//确定递归的传入参数为:存储节点数据的集合,和当前根节点,返回值为空
public void preOrder(List<Integer> list,TreeNode root){
if (root == null){
return;
}
//根
list.add(root.val);
//左
preOrder(list,root.left);
//右
preOrder(list,root.right);

}

迭代法:

借助栈这个数据结构,每次先将根节点入栈,处理完根节点,然后将其右孩子入栈,再将左孩子入栈,这样的出栈序列才是根左右

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
public List<Integer> preorderTraversal(TreeNode root) {
//结果数组
ArrayList<Integer> res = new ArrayList<>();
if (root == null) {
return res;
}

Stack<TreeNode> stack = new Stack<>();

//根节点入栈
stack.push(root);

while (!stack.isEmpty()) {
//弹出栈顶元素并存入结果数组
TreeNode node = stack.pop();
res.add(node.val);
//右节点不为空,入栈
if (node.right != null) {
stack.push(node.right);
}
//左节点不为空,入栈
if (node.left != null) {
stack.push(node.left);
}

}
return res;

}