C++实现的二叉树---家谱图

#include "iostream"
#include "string"
using namespace std;

/*******************************************************************************
树的节点类
*******************************************************************************/
class Node
{
friend class Person;
public:
Node()
{
name = '?';
lchild = NULL;
rchild = NULL;
}

private:
string name;
Node *lchild;
Node *rchild;
};

/*******************************************************************************
家谱成员类
*******************************************************************************/
class Person
{
public:
void Update(Person &L); //家谱成员改名
void Delete(Person &L); //删除部分家庭成员
void Insert(Person &L); //添加单个家庭成员
void Print(Node *p); //显示部分家庭成员
Node * Lookup(Node *p, string name); //查找某人
void Add(Person &L); //添加部分家庭成员
void Creat(Person &L); //创建家谱
Person()
{
root = NULL;
}

private:
Node *root;
};

/*******************************************************************************
创建家谱
*******************************************************************************/
void Person::Creat(Person &L)
{
cout << "请输入祖先的姓名：";
string rootname;
cin >> rootname;
Node *p = new Node;
p->name = rootname;
L.root = p;
cout << "此家谱的祖先是：" << p->name << '/n';
}

/*******************************************************************************
添加部分家庭成员
*******************************************************************************/
void Person::Add(Person &L)
{
cout << "请输入要建立家庭的人的姓名：";
string rootname;
cin >> rootname;
Node *s = Person::Lookup(L.root, rootname);
if (s) {
Node *r = s;
cout << "请输入" << s->name << "的儿女人数：";
int n;
cin >> n;
int m = n;
cout << "请依次输入" << s->name << "的儿女的姓名：";
while (m) {
Node *q = new Node;
string name;
cin >> name;
q->name = name;
if (m == n) {
s->lchild = q;
s = s->lchild;
}
else {
s->rchild = q;
s = s->rchild;
}
m --;
}
Person::Print(r);
}
else {
cout << "查无此人，请重新输入！" << '/n';
Person::Add(L);
}
}

/*******************************************************************************
查找某人
*******************************************************************************/
Node * Person::Lookup(Node *p, string name)
{
Node *t = NULL;
Node *s[100];
int top = 0;
while (p || top > 0) {
while (p) {
if (p->name == name) {
t = p;
}
s[++ top] = p;
p = p->lchild;
}
p = s[top --];
p = p->rchild;
}
return t;
}

/*******************************************************************************
显示部分家庭成员
*******************************************************************************/
void Person::Print(Node *p)
{
cout << p->name << "的第一代子孙是：" << p->lchild->name << '/t';
p = p->lchild;
while (p->rchild) {
cout << p->rchild->name << '/t';
p = p->rchild;
}
cout << '/n';
}

/*******************************************************************************
添加单个家庭成员
*******************************************************************************/
void Person::Insert(Person &L)
{
cout << "请输入要添加儿子（或女儿）的人的姓名：";
string rootname;
cin >> rootname;
Node *s = Person::Lookup(L.root, rootname);
if (s) {
Node *r = s;
cout << "请输入" << s->name << "新添加的儿子（或女儿）的姓名：" ;
Node *p = new Node;
string name;
cin >> name;
p->name = name;
if (!s->lchild) {
s->lchild = p;
}
else {
s = s->lchild;
while (s->rchild) {
s = s->rchild;
}
s->rchild = p;
}
Person::Print(r);
}
else {
cout << "查无此人，请重新输入！" << '/n';
Person::Insert(L);
}
}

/*******************************************************************************
删除部分家庭成员
*******************************************************************************/
void Person::Delete(Person &L)
{
cout << "请输入要解散家庭的人的姓名：";
string rootname;
cin >> rootname;
Node *s = Person::Lookup(L.root, rootname);
if (s) {
if (s->lchild) {
cout << "要解散家庭的人是：" << s->name << '/n';
Person::Print(s);
s->lchild = NULL;
}
else {
cout << s->name << "尚未有家庭！";
}
}
else {
cout << "查无此人，请重新输入！" << '/n';
Person::Delete(L);
}
}

/*******************************************************************************
家谱成员改名
*******************************************************************************/
void Person::Update(Person &L)
{
cout << "请输入要更改姓名的人的目前姓名：";
string rootname;
cin >> rootname;
Node *s = Person::Lookup(L.root, rootname);
if (s) {
cout << "请输入更改后的姓名：";
string name;
cin >> name;
s->name = name;
cout << rootname << "已更名为" << s->name << '/n';
}
else {
cout << "查无此人，请重新输入！" << '/n';
Person::Update(L);
}
}

/*******************************************************************************
主函数
*******************************************************************************/
void main()
{
cout<<"/n** 家谱管理系统 **"<> ch;
switch(ch) {
case 'A':
{
L.Add(L);
break;
}
case 'B':
{
L.Insert(L);
break;
}
case 'C':
{
L.Delete(L);
break;
}
case 'D':
{
L.Update(L);
break;
}
case 'E':
break;
default:
cout << "请输入正确的操作！" << '/n';
}
}
}