POJ 2003 Hire and Fire (Tree)

　　题目：Hire and Fire

　　题目翻译成数据结构就是：建树，加结点，删除结点，打印结点。只有删除结点稍微复杂点，因为删除设计掉树的调整。

　　首先要考虑树怎么存储才能使解题更顺手。

　　1.我们要存储每个结点的孩子，父亲和名字。存储孩子是因为第一个孩子可能会“升级”，存储父亲是因为要打印，名字肯定也是要打印的；

　　2.我们要知道每个结点的子树，删除结点会涉及调整；

　　3.我们要知道根节点，存放CEO。

　　所以我们的结点出来了，如下：

struct Tman
{
string name;
Tman *father;
list<Tman *> sons;
};

　思路都在注释里。

　　代码如下：

#include<iostream>
#include<map>
#include<list>
#include<string>
using namespace std;

//存储名字，父亲，孩子就行
struct Tman
{
string name;
Tman *father;
list<Tman *> sons;
};
//key是结点名字，value是名字的子树
map<string, Tman *> hash;

//根结点，指向CEO，打印是要用
Tman *root;

void print(long dep, Tman *now)
{
if(now == NULL) return;
for(long i = 1; i <= dep; ++i)
cout<<"+";
cout<<now->name<<endl;;
//递归打印每个孩子结点
for(list<Tman *>::iterator j = now->sons.begin(); j != now->sons.end(); ++j)
print(dep + 1, *j);
}

void hires(string n1, string n2)
{
Tman *boss = hash[n1];//父亲的子树指针
Tman *employee = new Tman();//新建一个Tman结构体，用于存储新加入的结点
employee->name = n2;//新加入结点的名字
employee->father = boss;//n1是n2的父亲
boss->sons.push_back(employee);//把n2放入n1的孩子list中
hash[n2] = employee;//新加入的结点也要有子树
}

void fire(string n1)
{
Tman *p = hash[n1];//指向n1结点的指针
hash.erase(n1);
while(p->sons.size() > 0)//如果要删的结点有孩子
{
p->name = p->sons.front()->name;//第一个孩子取代父亲的地位
hash[p->name] = p;//父亲的子树交给第一个孩子
p = p->sons.front();//p往下移动，始终指向孩子队列的第一个
}
p->father->sons.remove(p);//最后一个没有孩子的结点“删除”，实际上是上移了
delete p;//防止野指针
}

void solve()
{
string str1,str2;
long i;
cin>>str1;
root = new Tman();
hash[str1] = root;
root->name = str1;
while(cin>>str1)
{
if(str1 == "print")
{
print(0,root);
cout<<"------------------------------------------------------------"<<endl;
}
else if(str1 == "fire")
{
cin>>str2;
fire(str2);
}
else
{
cin>>str2;
cin>>str2;
hires(str1, str2);
}
}
}

int main()
{
solve();
return 0;
}