Leetcode99(恢复二叉搜索树)

给你二叉搜索树的根节点 root ，该树中的两个节点被错误地交换。请在不改变其结构的情况下，恢复这棵树。
示例 1：

输入：root = [1,3,null,null,2]
输出：[3,1,null,null,2]
解释：3 不能是 1 左孩子，因为 3 > 1 。交换 1 和 3 使二叉搜索树有效。
示例 2：

输入：root = [3,1,4,null,null,2]
输出：[2,1,4,null,null,3]
解释：2 不能在 3 的右子树中，因为 2 < 3 。交换 2 和 3 使二叉搜索树有效。

/**
 * 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:
    TreeNode *pre = nullptr;
    TreeNode *first = nullptr;
    TreeNode *second = nullptr;
    void dfs(TreeNode *root){
        // 判断
        if(!root) return;
        // 中序
        dfs(root->left);
        if(pre != nullptr){
            // 第一个节点，是第一个按照中序遍历时候前一个节点大于后一个节点，我们选取前一个节点。
            if(first == nullptr && pre->val >= root->val) first = pre;
            // 第二个节点，是在第一个节点找到之后，后面出现前一个节点大于后一个节点，我们选择后一个节点。
            if(first != nullptr && pre->val >= root->val) second = root;
        }
        pre = root;
        dfs(root->right);
    } 
    void recoverTree(TreeNode *root) {
        dfs(root);
        // 交换值
        int tmp = first->val;
        first->val = second->val;
        second->val = tmp;
    }
};

来源：力扣（LeetCode）
链接：link