POJ 2021 Relative Relatives（map+树的遍历）

题意：
　　今天是Ted的100岁生日。凑巧的是，他家族里面每个人都跟他同一天生日，但是年份不同。
　　现在只给出一些 父亲的名字，孩子的名字，以及孩子出生时父亲的年龄，
　　要求将Ted以外的家族成员按年龄降序排序，如果年龄相同，则按字母排序。

思路：遍历树。
　　根据题意，可通过父子关系建立一个有权无向树，Ted作为树的根节点。通过从Ted所在节点开始遍历树，给每个节点赋值。
　　最后根据每个成员的年龄排序即可。
　　这里用到map，建立名字到编号的映射，从而获取一个人在树中的节点编号。

#include <iostream>

#include <stdio.h>

#include <map>

#include <string>

#include <string.h>

#include <vector>

#include <algorithm>

using namespace std;

map<string,int> person;

int n;  //即题目中的X值

struct Line{

    string fname;  //父亲的名字

    string cname;  //孩子的名字

    int fage;  //孩子出生时，父亲的年龄

}line[];

struct Node{

    string name;  //名字

    int age;  //年龄

    int fage;  //出生时父亲的年龄，则该节点的年龄=父亲的年龄-fage

    vector<int> son;  //存储孩子的编号

    bool operator<(const Node tmp)const{

        if(age==tmp.age)

            return name<tmp.name;

        else

            return age>tmp.age;

    }

}node[];

//遍历，求节点的年龄

void dfs(int u){

    for(int i=;i<node[u].son.size();i++){

        int v=node[u].son[i];

        node[v].age=node[u].age-node[v].fage;

        dfs(v);

    }

}

int main()

{

    int t;

    scanf("%d",&t);

    for(int q=;q<=t;q++){

        printf("DATASET %d\n",q);

        person.clear();

        for(int i=;i<;i++)

            node[i].son.clear();

        int idx=;

        person["Ted"]=++idx;

        scanf("%d",&n);

        for(int i=;i<=n;i++){

            cin>>line[i].fname>>line[i].cname>>line[i].fage;

            //如果不存在该字符串的映射，即映射为0，则建立映射关系

            if(!person[line[i].fname])

                person[line[i].fname]=++idx;

            if(!person[line[i].cname]){

                person[line[i].cname]=++idx;

            }

        }

        //根节点

        node[].name="Ted";

        node[].age=;

        int u,v;

        for(int i=;i<=n;i++){

            u=person[line[i].fname];

            v=person[line[i].cname];

            node[u].name=line[i].fname;

            node[u].son.push_back(v);

            node[v].name=line[i].cname;

            node[v].fage=line[i].fage;

        }

        dfs();

        sort(node+,node+idx+);

        for(int i=;i<=idx;i++){

            if(node[i].name!="Ted")

                cout<<node[i].name<<" "<<node[i].age<<endl;

        }

    }

    return ;

}