L2-016 愿天下有情人都是失散多年的兄妹（DFS）

Posted 2022-04-06 MangataTS

题目链接

https://pintia.cn/problem-sets/994805046380707840/problems/994805061769609216

思路

我们把这个家族关系看成一颗二叉树，树的根节点是子元素，“左儿子” 是父亲，“右儿子” 是母亲，然后我们定义 $isman[i]$ 如果为 true 表示编号为id 的性别为男，反之为女，然后我们每次输入一行数据就连边，起点是当前编号，终点是父母编号（存在的话），然后我们在Q 次询问的时候，首先判断两个编号是否为异性，然后我们将以当前编号为起点的长度小于等于 $5$ 的所有的点全部放入一个 set中，另一个人同理，我们只需要遍历其中一个set然后在另一个set中查找是否存在当前的元素，即可判断是否5代内有交集，如果没有交集的话就输出Yes 否则输出No

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define mod 1000000007
#define endl "\\n"
#define PII pair<int,int>
#define INF 0x3f3f3f3f

const int N = 1e5+10;

int n,q;
vector<int> V[N];
bool isman[N];

void dfs(int id,int deep,set<int> &ans) 
	if(deep == 6) return;
	ans.insert(id);
	for(auto it : V[id])
		dfs(it,deep + 1,ans);


void slove(int l,int r)
	if(isman[l] == isman[r])
		cout<<"Never Mind"<<endl;
		return;
	
	set<int> S1,S2;
	dfs(l,1,S1);
	dfs(r,1,S2);
	bool fg = false;
	for(auto it : S1)
		if(S2.find(it) != S2.end()) 
			fg = true;
			break;
		
	
	if(fg)	cout<<"No"<<endl;
	else cout<<"Yes"<<endl;


int main()

	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n;
	int id,fa,ma;
	char sex;
	for(int i = 1;i <= n; ++i) 
		cin>>id>>sex>>fa>>ma;
		if(fa + 1) V[id].push_back(fa),isman[fa] = true;
		if(ma + 1) V[id].push_back(ma);
		if(sex == 'M') isman[id] = true;
		else isman[id] = false;
	
	cin>>q;
	int l,r;
	while(q--) 
		cin>>l>>r;
		slove(l,r);
	
	return 0;