首页 > 代码库 > Recover Binary Search Tree [leetcode]
Recover Binary Search Tree [leetcode]
本题是在中序遍历的基础上,找不合规范(不是递增)的树节点对,然后交换
首先看两种序列:
1. 1 3 2 4=>应该交换3和2
2. 4 3 2 1=>应交换4和1
这两种序列对应了不符合条件的BST的中序遍历序列的所有可能性---两个节点中序遍历相邻/不相邻
如果我们用一个数组swap保存所有中序遍历不递增的结果,那么这个数组只可能是2或者4的大小
而我们交换数组中节点对内容,只需交换第一个和最后一个树节点对的内容
vector<TreeNode *> swap; TreeNode * pre; void recoverTree(TreeNode *root) { pre = new TreeNode(-999999); swap = vector<TreeNode *>(); inorder(root); if (swap.size() > 1) { int val = swap[0]->val; swap[0]->val = swap[swap.size() - 1]->val; swap[swap.size() - 1]->val = val; } } void inorder(TreeNode * root) { if (root == NULL) return; inorder(root->left); if (pre->val > root->val) { swap.push_back(pre); swap.push_back(root); } pre = root; inorder(root->right); }由于只交换第一个和最后一个树节点内容,我们可以只保存第一个和最后一个Node
if(!first) first = pre;
last = root;
替换
swap.push_back(pre); swap.push_back(root);
代码如下:
TreeNode * first, * last; TreeNode * pre; void recoverTree(TreeNode *root) { pre = new TreeNode(-999999); first = last = NULL; inorder(root); if (first) { int val = first->val; first->val = last->val; last->val = val; } } void inorder(TreeNode * root) { if (root == NULL) return; inorder(root->left); if (pre->val > root->val) { if (!first) first = pre; last = root; } pre = root; inorder(root->right); }
Recover Binary Search Tree [leetcode]
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。