当前位置: 代码迷 >> 综合 >> leetcode 随笔 Recover Binary Search Tree
  详细解决方案

leetcode 随笔 Recover Binary Search Tree

热度:91   发布时间:2024-01-04 09:06:02.0

Recover Binary Search Tree

这个题很有意思,乍一看没什么思路,仔细想一想其实这是一道深度优先搜索的问题。

本身这棵树是按照中序遍历的从小到大生成的,那么我们在解析这颗树的时候如果利用dfs生成中序遍历的数列也应是从小到大依次排列好的。

题目已经给出了只有两对节点是需要互换的,如果我们把中序遍历的结果拿出来,以123456为例,会出现两种结果,第一种是123 54 6 这种连续的,另一种是 12 543 6这种不连续的互换节点。

class Solution {
public:TreeNode *s1=nullptr, *s2=nullptr, *last = new TreeNode(INT_MIN);void recoverTree(TreeNode* root) {bst(root);swap(s1->val, s2->val);}void bst(TreeNode *root){if (root == nullptr)return;bst(root->left);//print value;if (s1==nullptr&&last->val>=root->val){s1 = last;s2 = root;}if (s1 != nullptr&&last->val >= root->val){s2 = root;}last = root;bst(root->right);}
};
真是值得纪念的一道题啊(第100题)(* ̄︶ ̄)
  相关解决方案