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);
        }
    }
};