zoj 3811 Untrusted Patrol(BFS+并查集)

题目链接：zoj 3811 Untrusted Patrol

题目大意：给定n，m，k，表示有n个仓库，m条通道，k个传感器，现在给定n个传感器的位置和m条通道，现在要最这n个仓库进行巡逻，要求一次进过给定具有传感器的仓库，每个仓库经过的次数不限，单要求至少进过1次。

解题思路：首先判断是否为联通图，不连通的话肯定到不了。其次判断l是否等于k，如果不等于的话，说明至少有一个仓库到不了，剩下的就是考虑进过仓库的顺序，因为起点不限，所以第一个仓库的位置不会有问题，以第一个仓库为起点，做BFS，碰到有传感器的仓库将传感器关闭，并且将节点标记为1，表示说下一个位置可以直接到达该位置。然后考虑第二个位置，如果第二位置上的传感器没有关闭，则说明由第一个仓库到不了第二个仓库，否则重复操作，以第二个仓库为起点，做BFS

#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>

using namespace std;
const int maxn = 100005;

bool flag;
int N, M, C, t[maxn], v[maxn], f[maxn];
vector<int> g[maxn];

inline int getfar(int x) {
return x == f[x] ? x : f[x] = getfar(f[x]);
}

void init () {
scanf("%d%d%d", &N, &M, &C);
int x, a, b, n = N;

memset(t, 0, sizeof(t));
memset(v, 0, sizeof(v));
for (int i = 0; i <= N; i++) {
f[i] = i;
g[i].clear();
}

for (int i = 0; i < C; i++) {
scanf("%d", &x);
t[x] = 1;
}

for (int i = 0; i < M; i++) {
scanf("%d%d", &a, &b);

g[a].push_back(b);
g[b].push_back(a);

int p = getfar(a), q = getfar(b);
if (p != q) {
f[p] = q;
n--;
}
}

flag = (n == 1);
}

void bfs(int s) {
queue<int> que;
t[s] = 0; v[s] = 1;
que.push(s);

while (!que.empty()) {
int u = que.front();
que.pop();

for (int i = 0; i < g[u].size(); i++) {
int k = g[u][i];

if (v[k] || t[k]) {
v[k] = 1;
t[k] = 0;
continue;
}

v[k] = 1;
que.push(k);
}
}
}

bool judge () {
int n, x;
scanf("%d", &n);

if (n < C)
flag = false;

for (int i = 0; i < n; i++) {
scanf("%d", &x);
if (flag == false || (t[x] && i)) {
flag = false;
continue;
}
bfs(x);
}
return flag;
}

int main () {
int cas;
scanf("%d", &cas);
while (cas--) {
init();
printf("%s\n", judge() ? "Yes" : "No");
}
return 0;
}