力扣题目
解题思路
java代码
力扣题目:
给你二叉树的根节点 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
进阶:递归算法很简单,你可以通过迭代算法完成吗?
解题思路:
-
preorderTraversal
方法中:- 使用一个栈
stack
来辅助遍历,一个列表res
用于存储遍历结果。 - 首先判断根节点是否不为空,如果不为空则将其压入栈中。
- 然后进入一个循环,只要栈不为空:
- 弹出栈顶节点,并将其值添加到
res
中。 - 先将右子节点(如果存在)压入栈,再将左子节点(如果存在)压入栈。这样在后续的弹出操作中,就能先处理左子节点,再处理右子节点,符合前序遍历的顺序。
- 弹出栈顶节点,并将其值添加到
- 使用一个栈
java代码:
package org.example.mouth7.today23;
import java.util.ArrayList;
import java.util.List;
import java.util.Stack;
public class Leetcode144 {
public static void main(String[] args) {
TreeNode root = new TreeNode(1);
root.right = new TreeNode(2);
root.right.left = new TreeNode(3);
root.right.right = new TreeNode(4);
root.right.left.left = new TreeNode(5);
System.out.println(preorderTraversal(root));
}
public static List<Integer> preorderTraversal(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
List<Integer> res = new ArrayList<>();
if (root != null) {
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;
}
}
更多详细内容同步到公众号,感谢大家的支持!
每天都会给刷算法的小伙伴推送明日一题,并且没有任何收费项