文档讲解:二叉树的遍历
思路:
1、二叉树的递归遍历主要掌握递归的三步走,第一步是确定递归需要的参数和返回值,第二步确定递归的终止条件,第三步确定单层递归的逻辑。
2、二叉树的迭代遍历相较于递归遍历要难许多。在迭代遍历的时候,使用前中后序遍历统一的遍历方法,将访问的节点直接加入到栈中,但如果是处理的节点则后面放入一个空节点, 这样只有空节点弹出的时候,才将下一个节点放进结果集。
144. 二叉树的前序遍历(递归遍历)
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);
}
}
144. 二叉树的前序遍历(统一的迭代遍历)
/**
* 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 {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
Stack<TreeNode> stack = new Stack();
if(root!=null) stack.push(root);
while(!stack.isEmpty()){
TreeNode node = stack.peek();
if(node!=null){
stack.pop();
if(node.right!=null) stack.push(node.right);
if(node.left!=null) stack.push(node.left);
stack.push(node);
stack.push(null);
}else{
stack.pop();
node = stack.peek();
stack.pop();
res.add(node.val);
}
}
return res;
}
}