迭代(利用栈实现)
前序遍历为例,遍历顺序中->左->右,栈特点为先进后出。
所以放入顺序为中->取出中->右->左,这样左在下一步就成了中,以此类推。
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> result = new ArrayList<>();
if (root == null){
return result;
}
Stack<TreeNode> stack = new Stack<>();
//入栈
stack.push(root);
while (!stack.isEmpty()){
//从栈取出并放入结果集
TreeNode node = stack.pop();
result.add(node.val);
if (node.right != null){
stack.push(node.right);
}
if (node.left != null){
stack.push(node.left);
}
}
return result;
}
}
递归
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> result = new ArrayList<Integer>();
preorder(root, result);
return result;
}
public void preorder(TreeNode root, List<Integer> result) {
if (root == null) {
return;
}
result.add(root.val);//中
preorder(root.left, result);//左
preorder(root.right, result);//右
}
}