94. Binary Tree Inorder Traversal

二叉树的中序遍历

题目描述

中序遍历

思路

递归
题目还说递归太简单啦，你再用个迭代吧。

代码

• 递归

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> ans;
        if (root!=nullptr)
            Traversal(ans,root);
        return ans;
    }
    void Traversal(vector<int> & ans,TreeNode* root){
        if(root->left!=nullptr)
            Traversal(ans,root->left);
        ans.emplace_back(root->val);
        if(root->right!=nullptr)
            Traversal(ans,root->right);
    }
};

• 迭代

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector <int> ans;
        stack <TreeNode*> s;
        TreeNode* t=root;
        while(t!=nullptr||!s.empty()){
            if(t!=nullptr){
                if(t->left==nullptr){
                    ans.emplace_back(t->val);
                    t=t->right;
                }
                else{
                    s.emplace(t);//划重点：效率绝绝子，减少一次构造函数
                    t=t->left;
                }
            }            
            if(t==nullptr&&!s.empty()){
                t=&s.top();
                s.pop();
                ans.emplace_back(t->val);
                t=t->right;
            }
        }
        return ans;
    }
};

• 方法三：Morris 中序遍历
  其实整个过程我们就多做一步：假设当前遍历到的节点为 x，将 x 的左子树中最右边的节点的右孩子指向 x，这样在左子树遍历完成后我们通过这个指向走回了 x，且能通过这个指向知晓我们已经遍历完成了左子树，而不用再通过栈来维护，省去了栈的空间复杂度。
  感觉有点像线索树。

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        TreeNode *predecessor = nullptr;

        while (root != nullptr) {
            if (root->left != nullptr) {
                // predecessor 节点就是当前 root 节点向左走一步，然后一直向右走至无法走为止
                predecessor = root->left;
                while (predecessor->right != nullptr && predecessor->right != root) {
                    predecessor = predecessor->right;
                }
                
                // 让 predecessor 的右指针指向 root，继续遍历左子树
                if (predecessor->right == nullptr) {
                    predecessor->right = root;
                    root = root->left;
                }
                // 说明左子树已经访问完了，我们需要断开链接
                else {
                    res.push_back(root->val);
                    predecessor->right = nullptr;
                    root = root->right;
                }
            }
            // 如果没有左孩子，则直接访问右孩子
            else {
                res.push_back(root->val);
                root = root->right;
            }
        }
        return res;
    }
};