九度oj 题目1035：找出直系亲属

题目描述：

    如果A，B是C的父母亲，则A，B是C的parent，C是A，B的child，如果A，B是C的（外）祖父，祖母，则A，B是C的grandparent，C是A，B的grandchild，如果A，B是C的（外）曾祖父，曾祖母，则A，B是C的great-grandparent，C是A，B的great-grandchild，之后再多一辈，则在关系上加一个great-。

输入：

    输入包含多组测试用例,每组用例首先包含2个整数n（0<=n<=26）和m（0<m<50）, 分别表示有n个亲属关系和m个问题, 然后接下来是n行的形式如ABC的字符串，表示A的父母亲分别是B和C，如果A的父母亲信息不全，则用-代替，例如A-C,再然后是m行形式如FA的字符串,表示询问F和A的关系。
    当n和m为0时结束输入。

输出：

    如果询问的2个人是直系亲属，请按题目描述输出2者的关系，如果没有直系关系，请输出-。
    具体含义和输出格式参见样例.

样例输入：

3 2
ABC
CDE
EFG
FA
BE
0 0

样例输出：

great-grandparent
-

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <algorithm>
#define MIN -999999
int father[27];
int mother[27];

char temp[10];

int relation[27][27];

int find(int p, int q) {
    if(father[p] == q) {
        return 1;
    }
    if(mother[p] == q) {
        return 1;
    }
    if(relation[p][q] != MIN) {
        return relation[p][q];
    }
    if(father[p] != -1) {
        int ansf = find(father[p],q);
        if(ansf != 0) {
            relation[p][q] = 1 + ansf;
            return 1 + ansf;
        }   
    }
    if(mother[p] != -1) {
        int ansm = find(mother[p],q);
        if(ansm != 0) {
            relation[p][q] = 1 + ansm;
            return 1 + ansm;
        }
    }
    return 0;
    
}

void prinfop(int n) {
    if(n == 0) {
        puts("-");
    }
    else if(n == 1) {
        puts("parent");
    }
    else if(n == 2) {
        puts("grandparent");
    }
    else {
        for(int i = 2; i < n; i++) {
            printf("%s","great-");
        }
        puts("grandparent");
    }
}

void prinfoc(int n) {
    if(n == 0) {
        puts("-");
    }
    else if(n == 1) {
        puts("child");
    }
    else if(n == 2) {
        puts("grandchild");
    }
    else {
        for(int i = 2; i < n; i++) {
            printf("%s","great-");
        }
        puts("grandchild");
    }
}
int main(int argc, char const *argv[])
{
    int n, m;
    //freopen("input.txt","r",stdin);
    scanf("%d %d",&n, &m);
    while(!(n == 0 && m == 0)) {
        for(int i = 0; i < 30; i++) {
            father[i] = -1;
            mother[i] = -1;
            for(int j = 0; j < 30; j++) {
                relation[i][j] = MIN;
            }
        }
        for(int i = 0; i < n; i++) {
            scanf("%s",temp);
            if(temp[0] == '-') {
                continue;
            }
            int child = temp[0] - 'A';
            if(temp[1] != '-') {
                int fa = temp[1] - 'A';
                father[child] = fa;
                relation[child][fa] = 1;
            }
            if(temp[2] != '-') {
                int ma = temp[2] - 'A';
                mother[child] = ma;
                relation[child][ma] = 1;
            }
            
        }
        for(int i = 0; i < m; i++) {
            scanf("%s",temp);
            if(temp[0] == '-' || temp[1] == '-') {
                puts("-");
                continue;
            }
            int p1 = temp[0] - 'A';
            int p2 = temp[1] - 'A';
            int ans1 = find(p1, p2);
            if(ans1 == 0) {
                int ans2 = find(p2,p1);
                prinfop(ans2);
            }
            else {
                prinfoc(ans1);
            }
        }
        scanf("%d %d",&n, &m);

    }   
    return 0;
}

这道题用了二叉树的遍历和动态规划的思想，最重要的是一遍通过了(忽略第一次忘记注释freopen)，开心！