题目:二叉树的中序遍历
给定一个二叉树,返回它的中序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
答案:
1.递归:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
List<Integer> list = new ArrayList<>();
public List<Integer> inorderTraversal(TreeNode root) {
if(root == null) return list; //递归基,空树直接返回
inorderTraversal(root.left); //遍历左子树
list.add(root.val);
inorderTraversal(root.right); //遍历左子树
return list;
}
}
2.非递归
(1)沿着根节点p一直向左走,沿途将根节点入栈,直到p为空
(2)取出上一层根节点访问之,然后转向该根节点的右子树进行中序遍历
(3)如果堆栈和p都为空,则表示没有更多的子树需要遍历,此时结束外层循环,完成整棵树的遍历
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
TreeNode p = root;
while(p != null || !stack.empty()) {
while(p != null) {
stack.push(p);
p = p.left;
}
if(!stack.empty()) {
p = stack.pop();
list.add(p.val);
p = p.right;
}
}
return list;
}
}