标题 | 144. 二叉树的前序遍历 |
难度 | 简单 |
天数 | 第10天,第1/3题 |
数据结构 | 树 |
给你二叉树的根节点
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
进阶:递归算法很简单,你可以通过迭代算法完成吗?
以上内容来源:力扣(LeetCode)
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
- 前序遍历,前序遍历首先访问根结点然后遍历左子树,最后遍历右子树。
- 我们利用Stack先进后出的特性,来遍历我们的树
Stack<TreeNode> stack = new Stack<TreeNode>();
- 创建res用于存储返回数据
List<Integer> res = new ArrayList();
- 先判断树是否为空,如果是空树直接返回空list
if(root == null){
return res;
}
- 将数添加进栈stack中
//树放入栈中
stack.push(root);
- 循环遍历栈,如果栈空了停止循环
while(!stack.isEmpty()){
xxxxx
}
- 循环中 我们先取出树
- 根节点添加到list中
- 如果右树存在,右树先添加进栈中
- 然后左树存在,左树添加进栈中
- 这样下次循环,就先开始从左树中开始遍历赋值
TreeNode node = stack.pop();
res.add(node.val);
if(node.right != null){
stack.push(node.right);
}
if(node.left != null){
stack.push(node.left);
}
完整代码:
/**
* 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 {
//数据结构 第 10 天 1/3 树
public List<Integer> preorderTraversal(TreeNode root) {
//用于存储遍历的数据
List<Integer> res = new ArrayList();
//利用栈先进后出的特性,达到我们前序遍历目的
Stack<TreeNode> stack = new Stack<TreeNode>();
if(root == null){
return res;
}
//树放入栈中
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;
}
}