0
点赞
收藏
分享

微信扫一扫

【LeetCode刷题】94 二叉树的中序遍历 java

芷兮离离 2022-04-07 阅读 46

题目

给定一个二叉树的根节点 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);
    }
}
举报

相关推荐

0 条评论