题目链接:https://www.patest.cn/contests/gplt/L2-016
呵呵。大家都知道五服以内不得通婚,即两个人最近的共同祖先如果在五代以内(即本人、父母、祖父母、曾祖父母、高祖父母)则不可通婚。本题就请你帮助一对有情人判断一下,他们究竟是否可以成婚?
输入格式:
输入第一行给出一个正整数N(2 <= N <= 104),随后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
每个人的父亲和母亲是确定的,因此对于某一个人的祖先的族谱可以看成是一个倒着的二叉树;对于查询的两个点,分别bfs五层标记祖先,如果出现重复则不可以。坑点是给出的父亲和母亲的性别也是可以确定的,因此也要将其标记。
代码:
#include <iostream> #include <cstdio> #include <cmath> #include <cstring> #include <algorithm> #include <string> #include <vector> #include <queue> #include <stack> #include <set> #include <map> #define INF 0x3f3f3f3f #define lowbit(x) (x&(-x)) #define eps 0.00000001 #define pn printf("\n") using namespace std; typedef long long ll; int n, dy[100001]; int fl; struct node{ int d, s; int f, m; }p[100001]; set <int> st; void bfs(int s) { int a = s; memset(dy, 0, sizeof dy); dy[a] = 1; queue <int> que; que.push(a); while(!que.empty()) { a = que.front(); que.pop(); if(dy[a] > 5) continue; if(st.count(a)) fl = 1; st.insert(a); if(p[a].f != -1) { dy[p[a].f] = dy[a] + 1; que.push(p[a].f); } if(p[a].m != -1) { dy[p[a].m] = dy[a] + 1; que.push(p[a].m); } } } int main() { for(int i=0;i<100000;i++) { p[i].d = i; p[i].f = p[i].m = -1; } cin >> n; int d, f, m; char s[2]; for(int i=0;i<n;i++) { scanf("%d%s%d%d",&d,s,&f,&m); p[d].s = s[0] == 'M' ? 0 : 1; if(f != -1) p[f].s = 0; if(m != -1) p[m].s = 1; p[d].f = f; p[d].m = m; } int k; scanf("%d",&k); for(int i=0;i<k;i++) { int u,v; scanf("%d%d",&u,&v); if(p[u].s == p[v].s) printf("Never Mind\n"); else { fl = 0; st.clear(); bfs(u); bfs(v); printf(fl ? "No\n" : "Yes\n"); } } }