1386：打击犯罪(black)

1386：打击犯罪(black)
时间限制: 1000 ms         内存限制: 65536 KB
【题目描述】
某个地区有n(n<=1000)个犯罪团伙，当地警方按照他们的危险程度由高到低给他们编号为1-n，他们有些团伙之间有直接联系，但是任意两个团伙都可以通过直接或间接的方式联系，这样这里就形成了一个庞大的犯罪集团，犯罪集团的危险程度由集团内的犯罪团伙数量唯一确定，而与单个犯罪团伙的危险程度无关（该犯罪集团的危险程度为n）。现在当地警方希望花尽量少的时间（即打击掉尽量少的团伙），使得庞大的犯罪集团分离成若干个较小的集团，并且他们中最大的一个的危险程度不超过n/2。为达到最好的效果，他们将按顺序打击掉编号1到k的犯罪团伙，请编程求出k的最小值。
【输入】
第一行一个正整数n。接下来的n行每行有若干个正整数，第一个整数表示该行除第一个外还有多少个整数，若第i行存在正整数k，表示i，k两个团伙可以直接联系。
【输出】
一个正整数，为k的最小值。
【输入样例】
7
2 2 5
3 1 3 4
2 2 4
2 2 3
3 1 6 7
2 5 7
2 5 6
【输出样例】
1
【提示】
【提示】
输出1（打击掉犯罪团伙）

//示例代码
#include <iostream>
#include <cstdio>
using namespace std;
const int N=1005;
bool sign=false;
int n;// n个团伙
int f[N];// f[i]储存第i个人属于的团伙根
int a[N][N];// a[i][j]储存第i个团伙的第j个人
int s[N];// s[i]储存第i个团伙的危险程度

// 并查集的查找函数，返回x号元素所在集合的根
int find(int x){
	if(f[x]==x) return f[x];
	return f[x]=find(f[x]);
}

int main(){
	scanf("%d",&n);
    // 输入每个团伙之间的联系情况，同时初始化并查集数组和集合大小数组
	for(int i=1;i<=n;i++){
        scanf("%d",&a[i][0]);
		f[i]=i;s[i]=1;
		for(int j=1;j<=a[i][0];j++) scanf("%d",&a[i][j]);
	}
    // 从后往前遍历各个团伙，并且在团伙之间存在直接联系的条件下更新并查集，并且判断是否满足题目要求
	for(int i=n;i>=1;i--){
		if(sign) break;  // 如果已经找到解则退出循环
		for(int j=1;j<=a[i][0];j++)
			if(a[i][j]>i){  // 只对编号大于当前团伙编号的团伙进行考虑
				int tx=find(i);  // 获取当前团伙所在的集合的根节点
				int ty=find(a[i][j]);  // 获取与当前团伙有直接联系的团伙所在的集合的根节点
				if(tx!=ty){  // 如果两个团伙不在同一个集合中，则将它们所在的集合合并
					f[ty]=tx;
					s[tx]+=s[ty];// 第tx团伙的危险程度增加
                    // 判断是否达到题目要求，如果当前集合的大小超过了(n+1)/2，则输出解并且退出循环
					if(s[tx]>((n+1)/2)){
						printf("%d",i);
						sign=true;break;
					}
				}
			}
	}
    return 0;
}