leetcode 94.二叉树的中序遍历

leetcode 94.二叉树的中序遍历

题干

返回一棵二叉树的中序遍历

题解

递归的做法其实是在隐式维护一个栈，所以直接用栈来写内存消耗是基本一致的。
* 注意点在于两个循环

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> ans;
        stack<TreeNode*> S;
        while(!S.empty()||root!=nullptr)
        {
            while(root!=nullptr)
            {
                S.push(root);
                root = root->left;
            }//只要root不为空就在单个循环内将root为根最左子树的节点全部推入栈，如果root没有左树，仍然会将root推入栈，所以后续操作会继续取right
            root = S.top();
            ans.push_back(root->val);
            S.pop();
            root = root->right;
        }
        return ans;
    }
};

morris中序遍历

核心：
* 对于每个root 检查其是否有左子树 若有则找到左子树的最右节点 记为predecessor 然后将predecessor的右节点连接到root 并将root = root->left后重复此操作
* 这样操作下来，原先root的左子树所有以中序节点为根的左子树的最右节点都会和root连接
* 当root到最左节点时已无左子树，此时推入root->val然后使root = root->right 由于predecessor会与中序节点连接所以root会回到上一个中序节点
* 若中序节点的predecessor的right已不为空说明已经遍历过，则将中序节点推入然后root = root->right
* 核心：将一个左子树的最右节点的右孩子与子树的父节点连接
* 官方题解：https://leetcode-cn.com/problems/binary-tree-inorder-traversal/solution/er-cha-shu-de-zhong-xu-bian-li-by-leetcode-solutio/

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;
    }
};