LeetCode-144. 二叉树的前序遍历
144. 二叉树的前序遍历
给你二叉树的根节点 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
public List<Integer> preorderTraversal(TreeNode root) {
if (root == null) {
return Collections.emptyList();
}
Stack<TreeNode> stack = new Stack<>();
List<Integer> ans = new ArrayList<>();
stack.push(root);
Set<TreeNode> set = new HashSet<>();
while (!stack.isEmpty()) {
TreeNode node = stack.peek();
if (!set.contains(node)) {
ans.add(node.val);
set.add(node);
}
if (node.left != null && !set.contains(node.left)) {
stack.push(node.left);
continue;
}
stack.pop();
if (node.right != null) {
stack.push(node.right);
}
}
return ans;
}