题目
给定一个二叉树的根节点 root
,返回 它的 中序 遍历 。
示例
方法一:迭代
其实中序遍历的思想就是:左根右。 我们可以new一个链表res来保存弹栈元素最终输出,并且使用链栈来记录当前节点。
思想就是,从根节点开始,依次将其左孩子入栈,直至null。接着每弹栈一个元素,查看其是否有右子树,如果有,将其右子树依旧按照从右子树的根开始把左分支依次入栈。直至最后栈为空或者根为空。
/**
* 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) {
List<Integer> res = new ArrayList<Integer>();
Deque<TreeNode> stk = new LinkedList<TreeNode>();
while (root != null || !stk.isEmpty()) {
while (root != null) {
stk.push(root);
root = root.left;
}
root = stk.pop();
res.add(root.val);
root = root.right;
}
return res;
}
}
方法二:递归
按照 左子树,根节点,右子树的方式递归遍历这棵树。
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
inorder(root, res);
return res;
}
public void inorder(TreeNode root, List<Integer> res) {
if (root == null) {
return;
}
inorder(root.left, res);
res.add(root.val);
inorder(root.right, res);
}
}