当前位置: 代码迷 >> 综合 >> POJ 1611 - The Suspects
  详细解决方案

POJ 1611 - The Suspects

热度:4   发布时间:2023-12-24 11:23:39.0

题目大意:有编号为0~n-1的n个人,其中编号0中了SARS,如果跟他同一组就也会得病,问一共有多少人得病。输入m组,每组的成员。只有有同一个组员,则两个组是同一个组。

解题思路:并查集基础,只是因为0号得病,所以0号作为最上级的话便于查找,所以在合并时要考虑是否其中某个的上级为0,如果有,则合并时注意将另一个的上级的上级改为0。

ac代码:

#include <iostream>
using namespace std;
int n, m, a[30005], temp, pre[30005], sum;
int find(int x)
{int r=x, temp;while (r != pre[r])r = pre[r];while (r != x){temp = pre[x];pre[x] = r;x = temp;}
return r;
}
void join(int x, int y)
{int a=find(x), b=find(y);if (!b && a != b)pre[a] = b;else if (a != b)pre[b] = a;
}
int main()
{while (scanf("%d%d", &n, &m)!=EOF && n+m){sum = 1;for (int i=0; i<n; i++)pre[i] = i;for (int i=0; i<m; i++){scanf("%d", &temp);for (int j=0; j<temp; j++){scanf("%d", &a[j]);if (j)join(a[j-1], a[j]);}}for (int i=1; i<n; i++)if (!find(i))sum++;			printf("%d\n", sum);	}
return 0;
}