【DFS或BFS】之L2-016 愿天下有情人都是失散多年的兄妹 详解

 

题目链接： 

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

思路：

 利用的是二叉树的原理。把每个人的亲缘关系用一棵二叉树表示出来。然后对情侣的一方先进行遍历，并做好标记。接着对另一方进行遍历，如果遍历过程中发现亲属已经被标记过了，证明是五代之内的兄妹！

另外需要注意的是，要完善每个人父母信息，特别是性别！

示例二的递归思想更深入，需要慢慢品尝理解emmm

遍历采用DFS

具体看代码~~

示例一：这里利用了全局变量flag

#include<iostream>
#include<string.h>
using namespace std;
struct node
{
	char gender;//性别
	int f_m_id[2] = { -1,-1 };//父母的id
};
node cp[100000];
int book[100000] = { 0 };
int flag = 0;//全局变量
int dfs(int id, int time)
{
	if (time == 5 || id == -1 || flag == 1)//递归出口
		return 0;
	book[id] = 1;
	for (int i = 0; i <= 1; i++)
	{
		int f_id = cp[id].f_m_id[i];
		if (f_id == -1)
		{
			continue;
		}
		if (book[f_id] == 0)
		{
			dfs(f_id, time + 1);
		}
		else
		{
			flag = 1;//值不会每层函数更新
			return 0;
		}
	}
	if (flag == 1)
		return 0;
	return 1;
}
int main()
{
	int n, id, cnt, ch1, ch2;
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> id;
		cin >> cp[id].gender >> cp[id].f_m_id[0] >> cp[id].f_m_id[1];
		cp[cp[id].f_m_id[0]].gender = 'M';
		cp[cp[id].f_m_id[1]].gender = 'F';//补充其父母的性别
	}
	cin >> cnt;
	while (cnt--)
	{
		cin >> ch1 >> ch2;//两个情侣
		memset(book, 0, sizeof(book));
		flag = 0;
		if (cp[ch1].gender == cp[ch2].gender)
		{
			cout << "Never Mind" << endl;
		}
		else if (dfs(ch1, 0) && dfs(ch2, 0))//如果两者的父辈没有重叠的；假设重叠了，一个返回0，另一个返回1
		{
			cout << "Yes" << endl;
		}
		else
			cout << "No" << endl;
	}
	system("pause");
	return 0;
}

示例二：不需要全局变量

#include<iostream>
#include<string.h>
using namespace std;
struct node
{
	char gender;//性别
	int f_m_id[2] = { -1,-1 };//父母的id
};
node cp[100000];
int book[100000] = { 0 };
//int flag = 0;//全局变量
int dfs(int id, int time)
{
	int t;
	if (time == 5 )//递归出口
		return 1;
	book[id] = 1;
	for (int i = 0; i <= 1; i++)
	{
		int f_id = cp[id].f_m_id[i];
		if (f_id == -1)
		{
			continue;
		}
		if (book[f_id] == 0)
		{
			
			t=dfs(f_id, time + 1);
			
				
		}
		else
		{
			//flag = 1;//值不会每层函数更新
			return 0;
		}
		if (t == 0)
			break;
	}
	//if (flag == 1)
		//return 0;
	if (t == 0)
	{
		return t;
	}
	return 1;
}
int main()
{
	int n, id, cnt, ch1, ch2;
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> id;
		cin >> cp[id].gender >> cp[id].f_m_id[0] >> cp[id].f_m_id[1];
		cp[cp[id].f_m_id[0]].gender = 'M';
		cp[cp[id].f_m_id[1]].gender = 'F';//补充其父母的性别
	}
	cin >> cnt;
	while (cnt--)
	{
		cin >> ch1 >> ch2;//两个情侣
		memset(book, 0, sizeof(book));
		//flag = 0;
		if (cp[ch1].gender == cp[ch2].gender)
		{
			cout << "Never Mind" << endl;
		}
		else if (dfs(ch1, 0) && dfs(ch2, 0))//如果两者的父辈没有重叠的；假设重叠了，一个返回0，另一个返回1
		{
			cout << "Yes" << endl;
		}
		else
			cout << "No" << endl;
	}
	system("pause");
	return 0;
}