[LeetCode] 99. Recover Binary Search Tree

---

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

---

一棵二叉搜索树，有两个节点换过来了，需要找出这两个节点，换回来。

我们发现，一棵完好的BST，它的中序遍历是一个有序的数组n。

从左往右找到第一个n[i-1] >n[i], n[i-1]为第一个错误节点。
从右往左找到第一个n[i+1] < n[i], n[i+1]为第二个错误节点，也就是从左往右找到最后一个n[i-1]>n[i], n[i]为第二个错误节点。

---

void helper(TreeNode* root, vector<TreeNode*>& wrong_root, TreeNode* &last_root) {        if (root == NULL) return;        helper(root->left, wrong_root, last_root);        if (last_root && last_root->val > root->val) {            // 左边第一个错的肯定是错误节点            if (wrong_root[0] == NULL){                wrong_root[0] = last_root;            }            // 右边最后一个错的垦丁是错误节点            if (wrong_root[0] != NULL) {                wrong_root[1] = root;            }         }        last_root = root;        helper(root->right, wrong_root, last_root);    }    void recoverTree(TreeNode* root) {        vector<TreeNode*> wrong_root(2, 0);        TreeNode* last_root = NULL;        helper(root, wrong_root, last_root);        int tmp = wrong_root[0]->val;        wrong_root[0]->val = wrong_root[1]->val;        wrong_root[1]->val = tmp;    }