【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;
}