leetcode 99. 恢复二叉搜索树(中序遍历)

二叉搜索树中的两个节点被错误地交换。

请在不改变其结构的情况下,恢复这棵树。

示例 1:

输入: [1,3,null,null,2]

   1
  /
 3
  \
   2

输出: [3,1,null,null,2]

   3
  /
 1
  \
   2
示例 2:

输入: [3,1,4,null,null,2]

  3
 / \
1   4
   /
  2

输出: [2,1,4,null,null,3]

  2
 / \
1   4
   /
  3
进阶:

使用 O(n) 空间复杂度的解法很容易实现。
你能想出一个只使用常数空间的解决方案吗?

解法一:按照中序遍历将root遍历一遍,并将每个位置的地址和该节点的值分别保存到数组中。然后将保存节点值的数组从小到大排序,按地址的顺序赋给每个节点即可。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void recoverTree(TreeNode* root) {
        if(root == nullptr)
            return;
        vector vec;
        inorder(root,vec);
        sort(vec.begin(),vec.end());
        int k = 0;
        recover(root,vec,k);
    }
    void inorder(TreeNode* root,vector &vec){//vec要使用引用
        if(root == nullptr)
            return;
        inorder(root->left,vec);
        vec.push_back(root->val);
        inorder(root->right,vec);
    }
    void recover(TreeNode* root,vector vec,int &k){
        if(root == nullptr)
            return;
        recover(root->left,vec,k);
        root->val = vec[k++];
        recover(root->right,vec,k);
    }
};

解法二:参考链接https://blog.csdn.net/sinat_35261315/article/details/78885322

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    TreeNode *pre,*fir,*sec;
public:
    void recoverTree(TreeNode* root) {
        if(root == NULL)
            return;
        TreeNode minNode(INT_MIN);
        pre = &minNode;
        fir = NULL;
        sec = NULL;
        
        inorder(root);
        swap(fir->val,sec->val);
    }
    void inorder(TreeNode* root){//vec要使用引用
        if(root == NULL)
            return;
        
        inorder(root->left);

        //当上一个节点的值大于当前节点值时,说明上一个节点是错误的,影响了递增
        if(fir==NULL && pre->val>root->val)
            fir = pre;
        
        //当firstNode记录完后,当前节点小于前面的节点,说明当前节点是错误的
        if(fir!=NULL && pre->val>root->val)//注意这个判断.都是大于等于时为错,因为小于才是对的
            sec = root;
        pre = root; //prevNode记录上一个节点
        inorder(root->right);
    }
};

 

你可能感兴趣的:(leetCode,LeetCode)