Recover Binary Search Tree
原题链接Recover Binary Search Tree
给定一个二叉搜索树(BST),但是树中有两个节点被交换了,找到这两个节点,将其修正
二叉搜索树规则
- 当前节点的值大于左子树的所有节点值
- 当前节点的值小于右子树的所有节点值
- 左右子树同样是二叉搜索树
对于二叉搜索树,如果进行中序遍历,那么得到的序列一定是递增的。而对于本题而言,如果使用中序遍历,那么序列中一定会有两个值影响到了递增关系
那么,只需要改变这两个值的位置,然后重新中序遍历一遍为所有节点重新赋值即可,空间复杂度为O(n)
代码如下
/**
* 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){
vector<int> nums;
//先中序遍历把所有节点值找到
inOrder(root, nums);
//排序,修正节点值顺序
std::sort(nums.begin(), nums.end());
int i = 0;
//在中序遍历一遍,重新为节点赋值
recover(root, nums, i);
}
private:
void inOrder(TreeNode* root, vector<int>& nums)
{
if(!root) return;
inOrder(root->left, nums);
nums.emplace_back(root->val);
inOrder(root->right, nums);
}
void recover(TreeNode* root, vector&l