力扣工作周刷題 - 94. 二叉樹的中序遍歷

2020.10.18
原題：點選此處

經典二叉樹問題
時間複雜度O（n）
空間複雜度O（n）

法1：遞迴。（dfs）

class Solution {
    
    List<Integer> result = new ArrayList<>();

    public List<Integer> inorderTraversal(TreeNode root) {
        dfs(root);
        return result;
    }

    public void dfs(TreeNode root){
        if(root == null){
            return;
        }
        dfs(root.left);
        result.add(root.val);
        dfs(root.right);
        return;
    }
}

法2：迭代（人為用棧進行記錄）
可能需要多寫幾遍，特別是兩個限制條件很容易錯。

class Solution {
    
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        //兩個限制條件
        while(root != null || !stack.isEmpty() ){
            while(root != null){
                stack.add(root);
                root = root.left;
            }
            root = stack.pop();
            result.add(root.val);
            root = root.right;
        }
        return result;
    }

}