这个题很有意思,乍一看没什么思路,仔细想一想其实这是一道深度优先搜索的问题。
本身这棵树是按照中序遍历的从小到大生成的,那么我们在解析这颗树的时候如果利用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题)(* ̄︶ ̄)