【刷题之路】寻找错误结点

一棵二叉树原本是搜索二叉树,但是其中有两个节点调换了位置,使得这棵二叉树不再是搜索二叉树,请找到这两个错误节点并返回他们的值。保证二叉树中结点的值各不相同。

给定一棵树的根结点,请返回两个调换了位置的值,其中小的值在前。

二叉搜索树按照中序遍历应该是一个升序数组,如果有错位节点,包括两种情况

情况1:1 5 3 4 2 ,其中共有两次降序数字对,分别为5 3与4 2,只要取第一对的较大数字与第二对的较小数字即可

情况2:1 2 4 3 5,其中只有一对降序数字,则直接返回即可。

class FindErrorNode {
public:
    vector<int> findError(TreeNode* root) {
        // write code here
        vector<int> temp;
        vector<int> res;
        midtree(root,temp);
        int p1,p2;
        int i,j;
        for(i=0;i<temp.size()-1;i++){   
            if(temp[i]>temp[i+1]){ //出现第一对降序数字对,存下来
                p1=temp[i];
                p2=temp[i+1];
                for(j=i+1;j<temp.size()-1;j++){  //在之后寻找第二对降序数字对,更新第二个数为第二对中较小的数字
                    if(temp[j]>temp[j+1]){
                        p2=temp[j+1];
                    }
                }
                break; //跳出循环
            }
        }
        res.push_back(p2); //p2小于p1
        res.push_back(p1);
        return res;
    }
    void midtree(TreeNode* root,vector<int> &res){  //递归法中序遍历二叉树
        if(root==NULL) return;
        midtree(root->left,res);
        res.push_back(root->val);
        midtree(root->right,res);
    }
};

阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zyn2609530/article/details/51549303
文章标签: 刷题 二叉树 搜索
想对作者说点什么? 我来说一句

二叉查找树实现代码

2016年12月09日 9KB 下载

没有更多推荐了,返回首页

不良信息举报

【刷题之路】寻找错误结点

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭