题目链接:click~
/*题意:二叉搜索树种有两个结点被调换,要求将它们找出来并恢复*/
/**
*思路:中序遍历二叉树时,记录第一次违反有序规则的两个结点,这两个结点中的
* 第一个结点一定是被调换的结点,但是不能确定第二个结点是不是被调换的
* 另一个结点。继续寻找,如果第二次发现有违反有序规则的两个结点,那么
* 这两个结点中的第二个结点一定被调换的另一个结点
*/
class Solution {
public:
void recover(TreeNode *T, TreeNode *&pre, TreeNode *&n1, TreeNode *&n2) {
if(T == NULL) return;
//中序遍历
recover(T->left, pre, n1, n2);
//发现有违反规则的两个数
if(pre != NULL && pre->val > T->val) {
n2 = T; //第二次违反时,才是真正的n2
if(n1 == NULL) n1 = pre; //第一个被调换的结点肯定是第一次违反规则时的第一个结点
}
pre = T;
recover(T->right, pre, n1, n2);
}
void recoverTree(TreeNode *root) {
TreeNode *pre = NULL, *n1 = NULL, *n2 = NULL;
recover(root, pre, n1, n2);
if(n1 != NULL && n2 != NULL) {
swap(n1->val, n2->val);
}
}
};