题目描述
一棵二叉树原本是搜索二叉树,但是其中有两个节点调换了位置,使得这棵二叉树不再是搜索二叉树,请按升序输出这两个错误节点的值。(每个节点的值各不相同)
输入描述:
第一行输入两个整数 n 和 root,n 表示二叉树的总节点个数,root 表示二叉树的根节点。
以下 n 行每行三个整数 fa,lch,rch,表示 fa 的左儿子为 lch,右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子,rch同理)
ps:节点的编号就是该节点的值。
输出描述:
请按升序输出这两个错误节点的值。
示例1
输入
复制
3 1
1 2 3
2 0 0
3 0 0
输出
复制
1 2
思路:中序遍历数,如果出现了一次降序,那么相邻的两个数就是要找的数;如果出现了两次降序,那么第一个数是出现降序的第一个数,第二个数是第二次出现降序的第二个数。
注: 这里的中序遍历最好改成非递归写法。
#include<iostream>
using namespace std;
struct node
{
int lch,rch;
node(int x=0,int y=0):lch(x),rch(y){};//这里node构造函数必须给默认值
};
int a[500007];//储存中序遍历的顺序
int ct=0;//下标
node son[500007];//储存节点的儿子
void in_order(int root) //中序遍历
{
if(root==0)
return ;
in_order(son[root].lch);
a[ct++]=root;
in_order(son[root].rch);
}
int main()
{
int n,root;
scanf("%d%d",&n,&root);
for(int i=1;i<=n;i++)
{
int father,lch,rch;
scanf("%d%d%d",&father,&lch,&rch);
son[father].lch=lch,son[father].rch=rch;
}
in_order(root);
int mis[3]={0,0,0};
bool k=true;
for(int i=0;i<ct-1;i++)//这里必须时ct-1最后一位是不需要比较的
{
if(a[i]>a[i+1])
{
if(k)//第一次降序,要选择的是第一个数
{
mis[0]=i;
k=false;
}
else
{
mis[1]=i+1;
}
}
}
if(mis[1]==0)
cout<< a[ mis[0]+1 ]<<" "<< a[ mis[0] ]<<endl;
else
cout<< min( a[mis[0]] , a[mis[1]] )<<" "<< max( a[mis[0]] , a[mis[1]] )<<endl;
return 0;
}