【算法训练】LeetCode#94 二叉树的中序遍历

一、描述

94. 二叉树的中序遍历

给定一个二叉树的根节点 root ，返回 它的 中序 遍历 。

示例 1：

输入：root = [1,null,2,3]
输出：[1,3,2]

示例 2：

输入：root = []
输出：[]

示例 3：

输入：root = [1]
输出：[1]

二、思路

链表题做的差不多了，开始做做栈吧，不过这道题其实不用栈实现更容易一些....

三、解题

public static class Solution {
    // 无递归，栈实现
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        while (!stack.isEmpty() || root != null){
            if (root != null){
                stack.push(root);
                root = root.left;
            } else {
                root = stack.pop();
                ans.add(root.val);
                root = root.right;
            }
        }
        return ans;
    }

    // 递归遍历，直接入list
    public List<Integer> inorderTraversalV2(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        process(ans,root);
        return ans;
    }

    public void process(List<Integer> ans,TreeNode node){
        if (node == null){
            return;
        }
        process(ans,node.left);
        ans.add(node.val);
        process(ans,node.right);
    }
}