解析:(递归遍历)
题目要求:实现二叉树的中序遍历
大体思路:
二叉树深度优先遍历,常用的遍历方式:递归
递归三要素:
①参数和返回值:参数为TreeNode和记录结果的res,返回值为void
②终止条件:当遍历完叶子结点时终止,也就是TreeNode==null,return;
③确定单层的递归逻辑:中序遍历是左中右,先去遍历左子树,再保存根结点,最后遍历右子树。
代码:(递归遍历)
/**
* 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> inorderTraversal(TreeNode root) {
ArrayList<Integer> res = new ArrayList<>();
inorder(root, res);
return res;
}
public void inorder(TreeNode root, ArrayList<Integer> res) {
if (root == null) return;
inorder(root.left, res);
res.add(root.val);
inorder(root.right, res);
}
}
解析:(迭代法遍历)
大体思路:由于中序遍历是左中右,每次不是先处理根结点,所以不可以按照前序遍历的思路解题。
每次一直向左遍历并入栈,一直到cur为null,弹出栈中元素,cur改为弹出的元素的右孩子。
具体步骤:①当cur不为null,stack.push(cur); cur = cur.left;,一直向左遍历。
②cur为null时,出栈并将出栈的值保存到res数组中,并且 cur =出栈节点.right;转向右节点。
在执行②时,会碰到cur为null的两种情况,
情况1,如果cur是叶子节点的左孩子,那么弹出叶子节点,cur成为叶子节点的右孩子,右孩子为空,继续弹出,这次弹出的就是叶子节点的父节点,这样也就满足了左中右中的左中。
情况2,如果cur的父节点不是叶子节点,那么弹出父节点,cur会成为叶子节点的右孩子,这样就满足的左中右中的右
代码:(迭代法遍历)
/**
* 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> inorderTraversal(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
ArrayList<Integer> res = new ArrayList<>();
if (root == null) return res;
TreeNode cur = root;
// 有stack为空,但是cur不为空的情况
while (cur != null || !stack.isEmpty()) {
// 如果左孩子不为空,一直向左遍历
if (cur != null) {
stack.push(cur);
cur = cur.left;
} else {
// 如果cur为空,弹出元素,cur向右遍历
// 情况1,如果cur是叶子节点的左孩子,那么弹出叶子节点,cur成为叶子节点的右孩子,右孩子为空,继续弹出,这次弹出的就是叶子节点的父节点,这样也就满足了左中右中的左中。
// 情况2,如果cur的父节点不是叶子节点,那么弹出父节点,cur会成为叶子节点的右孩子,这样就满足的左中右中的右
cur = stack.pop();
res.add(cur.val);
cur = cur.right;
}
}
return res;
}
}