一棵二叉树原本是搜索二叉树,但是其中有两个节点调换了位置,使得这棵二叉树不再是搜索二叉树,请找到这两个错误节点并返回他们的值。保证二叉树中结点的值各不相同。
给定一棵树的根结点,请返回两个调换了位置的值,其中小的值在前。
思路:二叉搜索树按照中序遍历应该是一个升序数组,如果有错位节点,包括两种情况
情况1: 1 5 3 4 2,其中共有两次降序数字对,分别为5 3与4 2,只要取第一对的较大数字与第二对的较小数字即可
情况2: 1 2 4 3 5,其中只有一对降序数字,则直接返回即可。
中序遍历当前的二叉树,每次记录上一个数和当前的数进行比较,如果减小了就是出错的数,记第一次出错时的较大值(当前值的前面一个值)为n1,较小值(当前值)为n2,然后继续向下遍历,如果遇到第2次出错的情况,就用较小的值(当前值)替换n2,表示这才是与n1交换的值,最后显然n1,n2就是交换的2个值,并且n1>n2,由于要求返回时较小的值在前面,因此返回n2,n1。可以使用递归也可以不使用递归来解决问题。
总的来说第一个出错的数(大)是第一次减小时前面的数;第二个出错的数(小)是最后一次减小时后面的数。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class FindErrorNode {
public:
vector<int> findError(TreeNode* root) {
vector<int> recur;
vector<int> res;
inOrderRecur(root,recur);
int count=0,big,small;
for(int i=0;i<recur.size()-1;i++){
if(recur[i]>recur[i+1]){ //此时出现减小的结点
/*只有第一次出现减小的结点时才设置recur[i],之后再次遇到减小的情况时不再更新recur[i],只更新recur[i+1],
因此使用count==0表示是否是第一次遇到减小的情况,从而保证只有第一次出现减小情况时更新recur[i]*/
if(count==0){ //出现第一对降序数字对,存下来
big=recur[i];
small=recur[i+1];
count=1;//
}else{//寻找第二对降序数字对,它始终是最后一次减小时的后面的一个数
small=recur[i+1];//更新第二个数为第二对中较小的数字
}
//count++;
}
}
res.push_back(small);
res.push_back(big);
return res;
}
void inOrderRecur(TreeNode* root,vector<int> &vec) {
if(root==NULL) return;
inOrderRecur(root->left,vec);
vec.push_back(root->val);
inOrderRecur(root->right,vec);
}
};