99:Recover Binary Search Tree【树】【DFS】

题目链接：click~

/*题意：二叉搜索树种有两个结点被调换，要求将它们找出来并恢复*/

/**
 *思路：中序遍历二叉树时，记录第一次违反有序规则的两个结点，这两个结点中的
 *      第一个结点一定是被调换的结点，但是不能确定第二个结点是不是被调换的
 *      另一个结点。继续寻找，如果第二次发现有违反有序规则的两个结点，那么
 *      这两个结点中的第二个结点一定被调换的另一个结点
 */

class Solution {
public:
    void recover(TreeNode *T, TreeNode *&pre, TreeNode *&n1, TreeNode *&n2) {
        if(T == NULL) return;

        //中序遍历
        recover(T->left, pre, n1, n2);
        //发现有违反规则的两个数
        if(pre != NULL && pre->val > T->val) {
            n2 = T;   //第二次违反时，才是真正的n2
            if(n1 == NULL) n1 = pre; //第一个被调换的结点肯定是第一次违反规则时的第一个结点
        }

        pre = T;
        recover(T->right, pre, n1, n2);
    }

    void recoverTree(TreeNode *root) {
        TreeNode *pre = NULL, *n1 = NULL, *n2 = NULL;
        recover(root, pre, n1, n2);
        if(n1 != NULL && n2 != NULL) {
            swap(n1->val, n2->val);
        }
    }
};