【Leetcode】144.二叉树的前序遍历（非递归，Morris算法）

/*
 * @lc app=leetcode.cn id=144 lang=cpp
 *
 * [144] 二叉树的前序遍历
 */

// @lc code=start
/**
 * 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> preorderTraversal(TreeNode* root) {
        vector<int> res;

        TreeNode *temp;
        TreeNode *cur = root;
        while(cur != NULL)
        {
            // 左子树为空，搜索右子树
            if (cur->left == NULL)
            {
                res.push_back(cur->val);
                cur = cur->right;
            }
            else
            {
                // 在左子树上找最右结点
                temp = cur->left;
                while(temp->right != NULL && temp->right != cur)
                    temp = temp->right;
                // 第一次搜索到这个结点
                if (temp->right == NULL)
                {
                    // 先序遍历，把这个结点保存
                    res.push_back(cur->val);
                    // 记录一下回溯的位置
                    temp->right = cur;
                    // 从左边继续搜索
                    cur = cur->left;
                }
                // 第二次遍历
                else  // temp->right == cur
                {
                    // 还原指针
                    temp->right = NULL;
                    // 左边在第一次遍历已经搜索过，向右边搜索
                    cur = cur->right;
                }
            }
            
        }

        return res;
    }
};