leetcode99恢复二叉搜索树题解

leetcode99恢复二叉搜索树

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

输入：root = [1,3,null,null,2]
输出：[3,1,null,null,2]
解释：3 不能是 1 的左孩子，因为 3 > 1 。交换 1 和 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=new TreeNode(INT_MIN);
    TreeNode* firstNode,*secondNode;
    void recoverTree(TreeNode* root) {
        dfs(root);
        int temp=firstNode->val;
        firstNode->val=secondNode->val;
        secondNode->val=temp;
    }
    void dfs(TreeNode *root)
    {
        if(!root)return;
        dfs(root->left);
        if(firstNode==NULL&&pre->val>root->val)firstNode=pre;
        if(firstNode!=NULL&&pre->val>root->val)secondNode=root;
        pre=root;
        dfs(root->right);
    }
};

标签：leetcode99,right,TreeNode,val,题解,二叉,firstNode,root,left
来源： https://www.cnblogs.com/chen-ustc/p/15853686.html