要求按照先序遍历的顺序输出给定二叉树的叶结点。实验代码分析
时间: 2024-05-09 09:20:30 浏览: 65
输出二叉树中的叶子结点
5星 · 资源好评率100%
实验代码如下:
```java
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) return res;
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.empty()) {
TreeNode node = stack.pop();
if (node.right != null) stack.push(node.right);
if (node.left != null) stack.push(node.left);
if (node.left == null && node.right == null) res.add(node.val);
}
return res;
}
}
```
本题的核心是使用栈模拟先序遍历的过程,当遍历到叶子节点时,将其加入到结果集中。具体实现如下:
1. 首先判断根节点是否为空,如果为空则直接返回空结果集。
2. 创建一个栈,并将根节点入栈。
3. 当栈不为空时,循环执行以下操作:
a. 弹出栈顶元素。
b. 如果该节点有右子节点,则将右子节点入栈。
c. 如果该节点有左子节点,则将左子节点入栈。
d. 如果该节点是叶子节点(即左右子节点都为空),则将其加入到结果集中。
4. 返回结果集。
需要注意的是,本题是按照先序遍历的顺序输出叶结点,因此在遍历时先遍历左子树,再遍历右子树。
阅读全文