文章目录
树节点的定义
首先,给出树节点的定义,方便我们理解下面的算法:
public class TreeNode {
public int val;
public TreeNode left;
public TreeNode right;
public TreeNode(int x) {
val = x;
}
}
如上述代码所示,二叉树节点定义为TreeNode,其中
val,表示当前节点的值;
left,表示当前节点的左儿子节点;
right,表示当前节点的右儿子节点。
二叉树的前序遍历
给定一个二叉树,返回它的前序遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,2,3]
如上述所示,前序遍历的顺序为:根节点、左子树和右子树。
递归
class Solution {
public List preorderTraversal(TreeNode root) {
List ans = new ArrayList<>();
helper(root, ans);
return ans;
}
private void helper(TreeNode node, List ans) {
if (node != null) {
ans.add(node.val);
if (node.left != null) {
helper(node.left, ans);
}
if (node.right != null) {
helper(node.right, ans);
}
}
}
}
迭代
class Solution {
public List postorderTraversal(TreeNode root) {
List ans = new ArrayList<>();
if (root == null) return ans;
Stack stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
TreeNode curr = stack.pop();
ans.add(curr.val);
if (curr.right != null) {
stack.push(curr.right);
}
if (curr.left != null) {
stack.push(curr.left);
}
}
return ans;
}
}
<