99. 恢复二叉搜索树(C++)---中序遍历解法

题目详情

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

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

示例 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) 空间复杂度的解法很容易实现。
• 你能想出一个只使用常数空间的解决方案吗？

 

——题目难度:困难


 

---

-下面代码

class Solution {
public:
    void recoverTree(TreeNode* root) {
        vector<int> result;
        inOrder(root, result); //中序遍历
        sort(result.begin(), result.end());
        recover(root, result);
    }
    
    void inOrder(TreeNode* root, vector<int>& result) {
    	if (root == NULL) return ;
    	
    	inOrder(root->left, result);
    	result.push_back(root->val);
    	inOrder(root->right, result);
    }
    
    void recover(TreeNode* root, vector<int>& result) { //中序遍历
    	if (root == NULL) return ;
    	
    	recover(root->left, result);
    	if (root->val == result[0]) {
    		result.erase(result.begin());
    	} else {
    		root->val = result[0];
    		result.erase(result.begin());
    	}
    	recover(root->right, result);
    }
};