一、题目描述
二、算法分析说明与代码编写指导
与 0 直接处在同一个组的,都是非典疑似患者。这些疑似患者如果同时处于别的组,那么别的组的全体成员也是非典意思患者。
这种关系可以用并查集来刻画。
我们对每个组,都令从第二个元素开始的根节点(父节点)全部指向第一个元素。
设 0 所在的组为 G,G’ 为与 G 不同的组。设 x 和 x’ 分别为 G、G’ 中的任意元素,它们的父节点分别为 f(x)、f(x’)。分两种情况完成证明:
当 G’ 在 G 后输入时,如果 x’ 在 G 中出现过,则 x’ 也算疑似感染者,并且 f(x’) = f(x) = f(0)。所以对先于 0 所在的组后输入的任意元素 x’,只要 f(x’) = f(0),则 x’ 要被记入感染总人数中。
当 G’ 在 G 前输入时,如果 x’ 后来在 G 中也出现,即 x’ = x,那么 f(x) = f(x’) = f(0)。所以对先于 0 所在的组后输入的任意元素 x’,只要 f(x’) = f(0),则 x’ 也要被记入感染总人数中。
综上:对任意的 x,只要 f(x) = f(0),则 x 要记入感染总人数。
于是我们很快可以写出代码。
三、AC 代码
#include<cstdio>
#include<algorithm>
#pragma warning(disable:4996)
using namespace std;
template<size_t n> class union_find {
private:
unsigned root[n], rank[n];
public:
union_find<n>() { init(); }
union_find<n>(const bool& WannaInit) { if (WannaInit == true)init(); }
void init() {
fill(rank, rank + n, 1);
for (unsigned i = 0; i < n; ++i)root[i] = i;
}
unsigned find_root(const unsigned& v) {
unsigned r = v, t = v, u;
if (t == root[v])return v;
while (r != root[r]) { r = root[r]; }
while (t != r) { u = root[t], root[t] = r, t = u; }
return r;
}
void path_compression() { for (unsigned i = 0; i < n; ++i)find_root(i); }
void merge(const unsigned& u, const unsigned& v) {
unsigned fu = find_root(u), fv = find_root(v);
if (rank[fu] <= rank[fv]) { root[fu] = fv; if (rank[fu] == rank[fv])++rank[fv]; }
else { root[fv] = fu; }
}
void modify(const unsigned& u, const unsigned& v) { root[v] = u; if (rank[u] == rank[v])++rank[u]; }
unsigned _rank(const unsigned& v) { return rank[v]; }
size_t size() { return n; }
};
unsigned n, m, k, a, b, c = 1, f; union_find<30000> u(0);
int main() {
for (;;) {
u.init(); c = 1;
scanf("%u%u", &n, &m); if (n == 0 && m == 0)return 0;
for (unsigned h = 0; h < m; ++h) {
scanf("%u%u", &k, &a);
for (unsigned i = 1; i < k; ++i) { scanf("%u", &b); u.merge(a, b); }
}
f = u.find_root(0);
for (unsigned i = 1; i < n; ++i)if (u.find_root(i) == f)++c;
printf("%u\n", c);
}
}