寻找二叉树某个结点的父结点和左右子女结点

寻找二叉树某个结点的父结点和左右子女结点

【问题描述】

首先利用二叉树的前序遍历建立一棵二叉树，然后输入一个x值，在二叉树中搜索结点值为x的父结点和x的左、右子女结点，请分别编写三个二叉树类的成员函数，如果找到x的父结点和左、右子女结点 ，输出结点的数据值，若无父结点和子女结点则输出 "NULL”，二叉树类的成员函数声明可以参考下面:

BinTreeNode * Parent (BinTreeNode *root,T item); //返回item的父结点指针
BinTreeNode * LeftChild (BinTreeNode *root,T item); //返回item的左孩子指针
BinTreeNode * RightChild (BinTreeNode *root,T item); //返回item的右孩子指针

【输入形式】假设二叉树的结点为字符型，输入“#”表示空结点，输入有2行，第一行为建立二叉树用的字符串，具体见样例输入，二叉树见教材P197页，图5.13，第2行为一个要搜索的x值

【输出形式】输出有3行，第1行为x的父结点的数据值，第2行为x的左孩子结点的数据值，第3行为x的右孩子结点的数据值，如果父结点为空（例如根结点），或左、右孩子结点为空（例如叶子结点），则输出NULL

【样例输入】

ABD##EG###C#F##

B

【样例输出】

A

D

E

【样例说明】

注意特殊情况，例如只有一个根结点的二叉树，或者输入的x值在二叉树中不存在

其他测试数据：

输入：

A##

A

输出：

NULL

NULL

NULL

输入：

ABD##EG###C#F##

F

输出：

C

NULL

NULL

输入：

ABD##EG###C#F##

C

输出：

A

NULL

F

输入：

ABD##EG###C#F##

H

输出：

NULL

NULL

NULL

代码如下:

#include<iostream>

using namespace std;

template<class T>
struct BinTreeNode
{
	T data;
	BinTreeNode<T>* leftchild;
	BinTreeNode<T>* rightchild;
	BinTreeNode()
	{
		leftchild = NULL;
		rightchild = NULL;
	}
	BinTreeNode(T x)
	{
		data = x;
		leftchild = NULL;
		rightchild = NULL;
	}
};

template<class T>
class Tree
{
public:
	BinTreeNode<T>* root;
	Tree();
	~Tree();
	void CreateTree(BinTreeNode<T>*& t);
	void deleTree(BinTreeNode<T>* t);
	BinTreeNode<T>* Parent(BinTreeNode <T>* root, T item);       //返回item的父结点指针     
	BinTreeNode<T>* LeftChild(BinTreeNode<T>* root, T item);     //返回item的左孩子指针     
	BinTreeNode<T>* RightChild(BinTreeNode<T>* root, T item);   //返回item的右孩子指针
};

template<class T>
Tree<T>::Tree()
{
	root = NULL;
}

template<class T>
Tree<T>::~Tree()
{

}

template<class T>
void Tree<T>::CreateTree(BinTreeNode<T>*& t)
{
	T x;
	cin >> x;
	if (x == '#')
	{
		t = NULL;
		return;
	}
	t = new BinTreeNode<T>(x);
	if (t == NULL)
		return;
	CreateTree(t->leftchild);
	CreateTree(t->rightchild);
}

template<class T>
void Tree<T>::deleTree(BinTreeNode<T>* t)
{
	if (t == NULL)
		return;
	deleTree(t->leftchild);
	deleTree(t->rightchild);
	delete t;
}

template<class T>
BinTreeNode<T>* Tree<T>::Parent(BinTreeNode <T>* root, T item)
{
	BinTreeNode<T>* temp = NULL;
	if (root == NULL)
		return NULL;
	if (root->leftchild != NULL)
	{
		if (root->leftchild->data == item)
			return root;
	}
	if (root->rightchild != NULL)
	{
		if (root->rightchild->data == item)
			return root;
	}
	temp = Parent(root->leftchild, item);
	if (temp)
		return temp;
	temp = Parent(root->rightchild, item);
	if (temp)
		return temp;
	return NULL;
}

template<class T>
BinTreeNode<T>* Tree<T>::LeftChild(BinTreeNode<T>* root, T item)
{
	BinTreeNode<T>* temp;
	if (root == NULL)
		return NULL;
	if (root->data == item)
		return root->leftchild;
	temp = LeftChild(root->leftchild, item);
	if (temp)
		return temp;
	temp = LeftChild(root->rightchild, item);
	if (temp)
		return temp;
}

template<class T>
BinTreeNode<T>* Tree<T>::RightChild(BinTreeNode<T>* root, T item)
{
	BinTreeNode<T>* temp;
	if (root == NULL)
		return NULL;
	if (root->data == item)
		return root->rightchild;
	temp = RightChild(root->leftchild, item);
	if (temp)
		return temp;
	temp = RightChild(root->rightchild, item);
	if (temp)
		return temp;
}

int main()
{
	Tree<char> temp;
	temp.CreateTree(temp.root);
	char ch;
	cin >> ch;
	BinTreeNode<char>* a, *b, *c;
	a = temp.Parent(temp.root, ch);
	if (a)
		cout << a->data << endl;
	else
		cout << "NULL" << endl;
	b = temp.LeftChild(temp.root, ch);
	if (b)
		cout << b->data << endl;
	else
		cout << "NULL" << endl;
	c = temp.RightChild(temp.root, ch);
	if (c)
		cout << c->data << endl;
	else
		cout << "NULL" << endl;
	temp.deleTree(temp.root);
	return 0;
}