给你二叉树的根节点 root ,返回它节点值的 前序 遍历。
示例 1:
输入:root = [1,null,2,3]
输出:[1,2,3]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
示例 4:
输入:root = [1,2]
输出:[1,2]
示例 5:
输入:root = [1,null,2]
输出:[1,2]
提示:
- 树中节点数目在范围 [0, 100] 内
- -100 <= Node.val <= 100
进阶
: 递归算法很简单,你可以通过迭代算法完成吗?
递归
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
walk(root,res);
return res;
}
private void walk(TreeNode root, List<Integer> res) {
if(null == root) {
return ;
}
res.add(root.val);
walk(root.left, res);
walk(root.right, res);
}
}
迭代
进行迭代遍历的话 需要一个栈,前序遍历的话 顺序是中 左 右 . 因此需要将右节点入栈
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
while (null != root || ! stack.isEmpty()) {
while (null != root) {
res.add(root.val);
stack.add(root.right);
root = root.left;
}
root = stack.pop();
}
return res;
}
}