洛谷P2307迷宫

刷题之路-图论

洛谷P2307迷宫

传送门

题目描述

做题思路

并查集，只有有一组的两个点是相同祖先，就说明不符合。除此之外，还需要判断是否是同一棵树，因此，用num记录祖先的数量，若为1则符合。

代码

#include<iostream>
#include<map>
#include<set>
#include<vector>
#include<algorithm>
#include<cstring>
#include<string>
#include<queue>
#include<functional>
#include<cmath>
#include<cstdlib>
using namespace std;
const int inf = 0x3f3f3f3f;
const int mod = 1e9 + 7;
typedef pair<int, int> pii;
typedef long long ll;
const int N = 100005;
const int M = 200005;
int p[N];	//祖先
int get(int x)
{
    
    
	if (p[x] != x) p[x] = get(p[x]);
	return p[x];
}
int main()
{
    
    
	int u, v;
	cin >> u >> v;
	while (u != -1 && v != -1)
	{
    
    
		memset(p, 0, sizeof p);
		bool same = 0;
		int num = 0;	//祖先数量
		while (u != 0 && v != 0)
		{
    
    
			if (!p[u])	//若为0，表示未访问过，则祖先是自己
			{
    
    
				p[u] = u;
				num++;
			}
			if (!p[v])
			{
    
    
				p[v] = v;
				num++;
			}
			int a = get(u), b = get(v);
			if (a == b) same = 1;	//相同则说明祖先一样，已相连
			else num--;			//否则，两人祖先合二为一
			p[a] = b;
			cin >> u >> v;
		}
		if (same || num != 1) puts("0");
		else puts("1");
		cin >> u >> v;
	}
	return 0;
}



int main()
{
    
    
	init();
	int q;
	scanf("%d%d%d%d", &n, &m, &k, &q);
	for (int i = 0; i < m; i++)
	{
    
    
		int u, v;
		scanf("%d%d", &u, &v);
		insert(u, v);
		insert(v, u);
	}

	for (int j = 0; j < q; j++)
	{
    
    
		if (j == 0)	//第一次需要进行判断哪些点有传感器
		{
    
    
			for (int i = 0; i < k; i++)
			{
    
    
				scanf("%d", &a[i]);
				has[a[i]] = 1;
			}
		}
		else for (int i = 0; i < k; i++) scanf("%d", &a[i]);
		if (bfs(a[0])) puts("Yes");
		else puts("No");
	}
	return 0;
}