【并查集】关押罪犯(BSOJ2809)
Description
S城现有两座监狱,一共关押着N名罪犯,编号分别为1~N。他们之间的关系自然也极不和谐。很多罪犯之间甚至积怨已久,如果客观条件具备则随时可能爆发冲突。我们用“怨 气值”(一个正整数值)来表示某两名罪犯之间的仇恨程度,怨气值越大,则这两名罪犯之间的积怨越多。如果两名怨气值为 c 的罪犯被关押在同一监狱,他们俩之间会发生摩擦,并造成影响力为c的冲突事件。
每年年末,警察局会将本年内监狱中的所有冲突事件按影响力从大到小排成一个列表,然后上报到S城Z市长那里。公务繁忙的Z市长只会去看列表中的第一个事件的影响力, 如果影响很坏,他就会考虑撤换警察局长。
在详细考察了N名罪犯间的矛盾关系后,警察局长觉得压力巨大。他准备将罪犯们在两座监狱内重新分配,以求产生的冲突事件影响力都较小,从而保住自己的乌纱帽。假设只要处于同一监狱内的某两个罪犯间有仇恨,那么他们一定会在每年的某个时候发生摩擦。那么,应如何分配罪犯,才能使Z市长看到的那个冲突事件的影响力最小?这个最小值是多少?
Input
输入文件名为 prison.in。输入文件的每行中两个数之间用一个空格隔开。
第一行为两个正整数N和M,分别表示罪犯的数目以及存在仇恨的罪犯对数。
接下来的M行每行为三个正整数 aj,bj,cj,表示aj号和bj号罪犯之间存在仇恨,其怨气值为cj。数据保证1≤aj
Output
输出文件prison.out共1行,为Z市长看到的那个冲突事件的影响力。如果本年内监狱 中未发生任何冲突事件,请输出0。
Sample Input
4 6
1 4 2534
2 3 3512
1 2 28351
1 3 6618
2 4 1805
3 4 12884
Sample Output
3512
Hint
【样例说明】
罪犯之间的怨气值如下面左图所示,右图所示为罪犯的分配方法,市长看到的冲突事件 影响力是3512(由2号和3号罪犯引发)。其他任何分法都不会比这个分法更优。
【数据范围】
对于30%的数据有N≤15。
对于70%的数据有N≤2000,M≤50000。
对于100%的数据有N≤20000,M≤100000。
Thinking
这道题首先要想到并查集思路,这是解题的基础。首先考虑问题,如何安排罪犯才能做到最优决策?明显的可以看出,要尽量先安排仇恨值大的组合,将他们放到不同的监狱里。从大到小进行决策直到发现有某一组组合已经在同一个监狱里,那么输出他们的仇恨值就是最终的答案。现在问题又来了,第一组罪犯可以轻松地随意安排到两个监狱中,但是从第二组开始,两个人到底应该各自去哪一个监狱呢?
这时要改变思路,既然维护罪犯在哪一个监狱不方便,那么我们干脆维护某两个罪犯不在一个监狱好了。考虑到并查集的本职工作是维护某两点在一个集合,不能很好地处理不在一个集合的情况,所以我们要曲线救国,通过保存某个点的“敌人”集合来代表和他不在一个监狱的罪犯,间接地实现维护某两点不在一个集合的情况。在加入关系的时候进行判断,如果某两点已经在一个集合,说明他们无论如何也安排不到不同的两个监狱了,输出仇恨值即可。
至此算法框架基本浮现。
(P.s 用并查集保存两点不在一个集合还可以使用对立点的方式,即给每一个实点都建立一个与之对立的虚点。当读取到A与B不在同一集合的信息时,将A与B',B与A'分别连边,这样也可以表示A与B不在同一集合。判断过程与上面的方法相同,实现起来也很简单,大家不妨试一下)
Code
1 #include<cstdio> 2 #include<algorithm> 3 #include<cstring> 4 using namespace std; 5 6 struct edge 7 { 8 int x,y,v; 9 }cr[100050]; 10 bool cmp(edge a,edge b) {return a.v>b.v;} 11 int n,m; 12 int an[20050],pr[20050],en[20050]; 13 14 int get_father(int x) 15 { 16 if(pr[x]==x) return x; 17 pr[x]=get_father(pr[x]); 18 return pr[x]; 19 } 20 void Union(int x,int y) 21 { 22 int fx=get_father(x); 23 int fy=get_father(y); 24 if(fx!=fy) pr[fx]=fy; 25 } 26 27 int main() 28 { 29 scanf("%d%d",&n,&m); 30 int x,y,v; 31 for(int i=1;i<=m;i++) 32 { 33 scanf("%d%d%d",&cr[i].x,&cr[i].y,&cr[i].v); 34 } 35 sort(cr+1,cr+m+1,cmp); 36 memset(pr,0,sizeof(pr)); 37 memset(en,0,sizeof(en)); 38 for(int i=1;i<=n;i++) {pr[i]=i;en[i]=0;} 39 for(int i=1;i<=m;i++) 40 { 41 x=cr[i].x; 42 y=cr[i].y; 43 int fx=get_father(x); 44 int fy=get_father(y); 45 if(fx==fy) 46 { 47 printf("%d\n",cr[i].v); 48 return 0; 49 } 50 if(en[x]==0) en[x]=y; 51 else Union(y,en[x]); 52 if(en[y]==0) en[y]=x; 53 else Union(x,en[y]); 54 } 55 printf("0\n"); 56 return 0; 57 }