leetcode:Recover Binary Search Tree

又有好几天没有做题，这道题看了题目很久了，一直在思考，因为要求常数空间而实际上递归解法需要的空间并不是常数级别的。看到网上的解答，有人说可以用Morris方法遍历二叉树，这样可以在o(1)空间复杂度和o(n)时间复杂度完成二叉树的遍历。但是好几天没有刷题了，想要快点做出这道题，所以还是用了递归的解法。Morris方法下次再做吧~

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

Recover the tree without changing its structure.

Note:
A solution using O( n) space is pretty straight forward. Could you devise a constant space solution?

confused what "{1,#,2,3}" means? > read more on how binary tree is serialized on OJ.

Show Tags

Have you met this question in a real interview? 

Yes

 

No

题目地址：https://oj.leetcode.com/problems/recover-binary-search-tree/

解题思路：思路是先得到二叉树的中序遍历，找出顺序不对的数，如果只有一个地方顺序不对交换两个数；如果出现两个顺序不对的地方，把第一个地方的较大的数和第二个地方较小的数交换。用change这个vector来存储顺序不对的节点。不管最后change中是只有两个节点还是有4个节点，都是交换第一个节点和最后一个节点的值。

代码：

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *pre;
    vector<TreeNode *> change;
    void recoverTree(TreeNode *root) {
        //思路是先得到二叉树的中序遍历，找出顺序不对的数，如果只有一个地方顺序不对交换两个数；如果出现两个顺序不对的地方，把第一个地方的较大的数和第二个地方较小的数交换
        if(root==NULL) return;
        if(root->left==NULL&&root->right==NULL) return;
        
        pre=NULL;
        change.clear();
        inorder(root);
        
        //交换change第一个节点和最后一个节点的值
        if(change.size()>0){
            int tmp=change[0]->val;
            change[0]->val=change[change.size()-1]->val;
            change[change.size()-1]->val=tmp;
        }
    }
    
    void inorder(TreeNode *root){
        if(root == NULL) return;
        
        inorder(root->left);
        if(pre == NULL){
            pre = root;
        }else{
            if(pre->val>root->val){
                change.push_back(pre);
                change.push_back(root);
            }
            pre = root;
        }
        inorder(root->right);
    }
};