POJ 2021 Relative Relatives（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[105];
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[105];

//遍历，求节点的年龄
void dfs(int u){
    for(int i=0;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=1;q<=t;q++){
        printf("DATASET %d\n",q);
        person.clear();
        for(int i=1;i<105;i++)
            node[i].son.clear();

        int idx=0;
        person["Ted"]=++idx;
        scanf("%d",&n);
        for(int i=1;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[1].name="Ted";
        node[1].age=100;
        int u,v;
        for(int i=1;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(1);
        sort(node+1,node+idx+1);
        for(int i=1;i<=idx;i++){
            if(node[i].name!="Ted")
                cout<<node[i].name<<" "<<node[i].age<<endl;
        }
    }
    return 0;
}