Leetcode144:二叉树前序遍历,morris遍历法

今天做了二叉树的前序遍历，题目要求是用迭代来做不用递归，因为迭代的效率比递归要高而且这也是很多公司面试题的一个要求。
第一遍解题用的是栈来模仿递归
代码：

/**
 * 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> preorderTraversal(TreeNode* root) {
        vector<int>res;
        if(root==NULL)
        return res;
        stack<TreeNode*>s;
        TreeNode *p=root;
        //结束条件是节点和栈同时为空，表示树已经遍历完成
        while(p||!s.empty())
        {
            //前序遍历是根左右，直接遍历到最左节点，同时保存根节点的值
            while(p)
            {
                s.push(p);
                res.emplace_back(p->val);
                p=p->left;
            }
            //每次拿出栈顶保存的节点，把当前节点p指向子树根节点的右孩子，进行下一次遍历
            if(!s.empty())
            {
                TreeNode*node=s.top();
                p=node->right;
                s.pop();
            }
        }
        return res;
    }
};

下面这张图是这个算法最好的执行结果

做完之后就去看看官方题解里有没有更好的思路，果然找到了一个Morris遍历，它的核心思想是利用树的空闲指针来实现空间的极限缩减，这让空间复杂度降为O(1);
Morris算法有两个原则:
1、如果cur无左孩子，cur向右移动
2、如果cur有左孩子，找到cur左子树的最右的节点，记为mostright
(1)如果mostright的right指向空，让其指向cur，cur向左移动
(2)如果mostright的right指向cur，让其指向空，cur向右移动

建议自己画一棵树去走一遍。
代码:

/**
 * 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> preorderTraversal(TreeNode* root) {
        vector<int>res;
        if(nullptr==root)
        return res;
        TreeNode *cur=root;
        TreeNode*mostright=nullptr;
        while(cur)
        {
            //判断到当前根节点的左子树是否存在
            mostright=cur->left;
            if(mostright!=nullptr)
            {
                //存在就执行第二条原则，找到左子树的最右边的节点，判断最右边的节点的右孩子指向
                while(mostright->right&&mostright->right!=cur)
                {
                    mostright=mostright->right;
                }
                if(mostright->right==nullptr)
                {
                    mostright->right=cur;
                    res.emplace_back(cur->val);
                    cur=cur->left;
                    continue;
                }
                else
                {
                    mostright->right=nullptr;
                }
            }
            else
            {
                res.emplace_back(cur->val);
            }
            cur=cur->right;
        }
        return res;
    }
};

最好的执行结果