洛谷：P2189 小Z的传感器（并查集染色）

洛谷：小Z的传感器

此题考察思维 —— 结合题意图性质对 并查集 的巧妙运用

一开始想着对于给定顺序的房间序号，从起点（顺序最前）跑到终点（顺序最后），观察是否能不重复地跑遍这个序列

但发现很难维护，不太能用最短路算法求解

看了题解 之后发现，可以用 扩散 的思想去模拟光顾过程

• 首先先标记没有传感器的房间，这些房间不可以被访问（因为有先后限制），其余未标记的房间显然都可以随意走，所以可以直接连接成一个连通块
• 每次按照序列先后，解除当前房间的标记，表示当前房间可以访问了（满足先后限制），然后把这房间可以走到的所有房间 （这些房间必须是没有标记过的，被标记的话就不能访问，也相当于必须要比该房间访问得晚） 连通（并查集处理）
• 随时判断序列中相邻前后两个房间是否连通了，若否，代表连通状态断开，给定的序列顺序肯定不存在

代码：

#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define sca scanf
#define pri printf
#define ul (u << 1)
#define ur (u << 1 | 1)
#define fx first
#define fy second
//#pragma GCC optimize(2)
//[博客地址](https://blog.csdn.net/weixin_51797626?t=1) 
using namespace std;

typedef long long ll;
typedef pair<int, int> PII;
typedef pair <int, PII> PI;

const int N = 100010, M = 400010, MM = N;
int INF = 0x3f3f3f3f, mod = 100003;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int n, m, k, T, S, D;
int h[N], e[M], ne[M], idx;
int p[N], seq[N];
bool st[N];

int find(int x) {
	if (p[x] != x)p[x] = find(p[x]);
	return p[x];
}

void add(int a, int b) {
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

int main() {
	cinios;

	int q;
	cin >> n >> m >> k >> q;
	mem(h, -1);

	for (int i = 0; i < m; i++) {
		int a, b;
		cin >> a >> b;
		add(a, b), add(b, a);//无向图
	}

	while (q--)
	{
		mem(st, 1);
		for (int i = 1; i <= n; i++)p[i] = i;

		for (int i = 1; i <= k; i++) {
			cin >> seq[i];
			if (i > 1)st[seq[i]] = false;//第一个房间不管如何都可以访问
		}

		for (int i = 1; i <= n; i++)
			if (st[i])//初始未标记房间连接成块
				for (int k = h[i]; ~k; k = ne[k]) {
					int j = e[k];
					if (st[j] && find(i) ^ find(j))
						p[find(i)] = find(j);
				}

		bool flag = true;
		for (int g = 2; g <= k; g++) {
			int k = seq[g];
			st[k] = true;//解除当前房间标记
			for (int i = h[k]; ~i; i = ne[i]) {
				int j = e[i];
				if (st[j] && find(k) ^ find(j))
					p[find(k)] = find(j);
			}

			if (find(k) ^ find(seq[g - 1])) { //查询连通状况
				flag = false;
				break;
			}
		}

		if (flag)cout << "Yes";
		else cout << "No";
		cout << '\n';
	}

	return 0;
}