解题思路
将有两个结点出错的二叉搜索树恢复为正确的二叉搜索树。用pre保存先序遍历的上一个结点,用a和b保存要交换的两个结点,若pre不为空且pre的值大于等于当前根节点的值,则修改b的值,当a为空修改a的值,并将pre修改为当前根结点,之后先序遍历根结点的右子树,最后若a、b不为空,则交换a和b的值。
代码
class Solution {
public:void preorder(TreeNode *root,TreeNode *&pre,TreeNode *&a,TreeNode *&b){if(root != NULL){preorder(root->left,pre,a,b);if(pre && pre->val >= root->val){if(!a)a = pre;b = root;}pre = root;preorder(root->right,pre,a,b);}}void recoverTree(TreeNode *root) {TreeNode *pre = NULL,*a = NULL,*b = NULL;preorder(root,pre,a,b);if(a && b){int tmp = a->val;a->val = b->val;b->val = tmp;}}
};