PTA - dfs

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PTA - dfs相关的知识,希望对你有一定的参考价值。

呵呵。大家都知道五服以内不得通婚,即两个人最近的共同祖先如果在五代以内(即本人、父母、祖父母、曾祖父母、高祖父母)则不可通婚。本题就请你帮助一对有情人判断一下,他们究竟是否可以成婚?

输入格式:

输入第一行给出一个正整数N(2 N 10?4??),随后N行,每行按以下格式给出一个人的信息:

本人ID 性别 父亲ID 母亲ID

其中ID是5位数字,每人不同;性别M代表男性、F代表女性。如果某人的父亲或母亲已经不可考,则相应的ID位置上标记为-1

接下来给出一个正整数K,随后K行,每行给出一对有情人的ID,其间以空格分隔。

注意:题目保证两个人是同辈,每人只有一个性别,并且血缘关系网中没有乱伦或隔辈成婚的情况。

输出格式:

对每一对有情人,判断他们的关系是否可以通婚:如果两人是同性,输出Never Mind;如果是异性并且关系出了五服,输出Yes;如果异性关系未出五服,输出No

输入样例:

24
00001 M 01111 -1
00002 F 02222 03333
00003 M 02222 03333
00004 F 04444 03333
00005 M 04444 05555
00006 F 04444 05555
00007 F 06666 07777
00008 M 06666 07777
00009 M 00001 00002
00010 M 00003 00006
00011 F 00005 00007
00012 F 00008 08888
00013 F 00009 00011
00014 M 00010 09999
00015 M 00010 09999
00016 M 10000 00012
00017 F -1 00012
00018 F 11000 00013
00019 F 11100 00018
00020 F 00015 11110
00021 M 11100 00020
00022 M 00016 -1
00023 M 10012 00017
00024 M 00022 10013
9
00021 00024
00019 00024
00011 00012
00022 00018
00001 00004
00013 00016
00017 00015
00019 00021
00010 00011

输出样例:

Never Mind
Yes
Never Mind
No
Yes
No
Yes
No
No

题目分析 : 题目给你一个家谱图,再给你一些询问点,查询所问的两个人是否可以通婚。
有两个坑的地方 :
  1 . 如果查询的人是不存在家谱中的,那么就要直接输出 No
  2 . 题目数据输入的时候性别只是对孩子而言的,但是数据查询的时候有可能不只查询孩子层间的关系,有可能会询问你父母间的关系,因为我们标记性别的时候同时也要将父母的性别标记上
代码示例 :
const int maxn = 1e5+5;
const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;

int n;
int f[maxn];
int vis[maxn]; // 1 M - -1 F
vector<int>ve[maxn];
int dep[maxn];

void dfs(int x, int d){
    if (d > 5) return;
    dep[x] = d;
    //printf("x = %d \n", x); 
    for(int i = 0; i < ve[x].size(); i++){
        int to = ve[x][i];
        dfs(to, d+1);
    }
}

int sign;
void dfs2(int x, int d){
    if (d > 5) return;
    if(sign) return;
    
    //printf("*** x = %d \n", x);
    if(dep[x]) {sign = 1; return;}
    
    for(int i = 0; i < ve[x].size(); i++){
        int to = ve[x][i];
        dfs2(to, d+1);
    }
}

int main() {
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    char s[5];
    int x, a, b, k;
    
    cin >> n;
    for(int i = 0; i <= n; i++) f[i] = i;
    memset(vis, 0, sizeof(vis));
    for(int i = 1; i <= n; i++){
        scanf("%d%s%d%d", &x, s, &a, &b);    
        if (s[0] == ‘M‘) vis[x] = 1;
        else vis[x] = -1;
        if (a != -1) vis[a] = 1;
        if (b != -1) vis[b] = -1;
        if (a != -1) ve[x].push_back(a);
        if (b != -1) ve[x].push_back(b);
        //printf("c = %c \n", s[0]);
    }
    
    cin >> k;
    //printf("%d %d\n", vis[21], vis[24]);
    for(int i = 1; i <= k; i++){
        scanf("%d%d", &a, &b);
        if (vis[a]*vis[b] > 0) {printf("Never Mind\n"); continue;}
        else if (vis[a]*vis[b] == 0) {printf("No\n"); continue;}
        memset(dep, 0, sizeof(dep));
        sign = 0;
        dfs(a, 1);
        dfs2(b, 1);
        
        if (sign) printf("No\n");
        else printf("Yes\n");        
    }
    return 0;
}

 

以上是关于PTA - dfs的主要内容,如果未能解决你的问题,请参考以下文章

PTA - dfs

PTA 1004 Counting Leaves (30)(30 分)(建树dfs或者bfs,未AC 段错误)

PTA 7-35 城市间紧急救援(Dijkstra+DFS)

PTA天梯赛L2

[code] PTA 胡凡算法笔记 DAY039

PTA7-3 子集和问题 (30 分) 回溯法