给定一个二叉树,返回它的中序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
遍历顺序为左中右
递归:从根节点开始,一直遍历它的左节点,直到节点为空时返回上一层,当前节点也就是中间节点,然后添加当前节点值,继续遍历该节点的右节点。
public static List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
inorder(root, res);
return res;
}
public static void inorder(TreeNode root, List<Integer> res) {
if (root == null) return;
if (root.left != null) inorder(root.left, res);
res.add(root.val);
if (root.right != null) inorder(root.right, res);
}
迭代:先遍历后往结果中加值可以使用栈来存储开始遍历到的节点,退出条件是现在要遍历的节点为空并且栈为空,当前节点不为空的话就添加到栈中,遍历它的左节点,直到为空返回,然后弹出当前栈顶节点,说明当前节点也就是中间节点,然后再切换到它的右节点,再次进行循环遍历,一个节点只能push到栈一次。
public static List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
TreeNode p = root;
while (p != null || !stack.empty()) {
while (p != null) {
stack.push(p);
p = p.left;
}
p = stack.pop();
res.add(p.val);
p = p.right;
}
return res;
}