HDU 6109 数据分割 【并查集+set】 （2017"百度之星"程序设计大赛 - 初赛（A））

题目链接：

　　http://acm.hdu.edu.cn/showproblem.php?pid=6109


题目大意：

　　给一些等式或不等式，xi=xj或xi!=xj，求一个划分，使得每一部分内的等式组去掉最后一个式子后成立。

题目思路：

　　【并查集+set】

　　开一个set Qi记录当前与这个元素Xi所在集合Si不相等的编号。

　　若条件为相等，判断其中一个元素Xi所在集合Si是否在另一元素Xj不相等集合Qj的set中，不在则将两个元素并成一个集合，并将不等的元素集合也合并（小的并到大的），否则不合法

　　若条件为不等，则判断这两个元素是否属于同一集合，是的话不合法，否则将对方加入到自己的set 中。

1 /****************************************************
2
3     Author : Coolxxx
4     Copyright 2017 by Coolxxx. All rights reserved.
5     BLOG : http://blog.csdn.net/u010568270 6
7 ****************************************************/
8 #include<bits/stdc++.h>
9 #pragma comment(linker,"/STACK:1024000000,1024000000")
10 #define abs(a) ((a)>0?(a):(-(a)))
11 #define lowbit(a) (a&(-a))
12 #define sqr(a) ((a)*(a))
13 #define mem(a,b) memset(a,b,sizeof(a))
14 const double EPS=0.00001;
15 const int J=10;
16 const int MOD=100000007;
17 const int MAX=0x7f7f7f7f;
18 const double PI=3.14159265358979323;
19 const int N=100004;
20 const int M=1004;
21 using namespace std;
22 typedef long long LL;
23 double anss;
24 LL aans;
25 int cas,cass;
26 int n,m,lll,ans;
27 int fa
,sz
;
28 set<int>a
;
29 int zhao(int aa)
30 {
31     if(fa[aa]==aa || fa[aa]==0)return aa;
32     return fa[aa]=zhao(fa[aa]);
33 }
34 bool Find(set<int>q,int x)
35 {
36     set<int>::iterator it;
37     for(it=q.begin();it!=q.end();it++)
38     {
39         if(zhao(*it)==x)return 1;
40     }
41     return 0;
42 }
43 int main()
44 {
45     #ifndef ONLINE_JUDGE
46     freopen("1.txt","r",stdin);
47 //    freopen("2.txt","w",stdout);
48     #endif
49     int i,j,k;
50     int x,y,z,fx,fy;
51 //    for(scanf("%d",&cass);cass;cass--)
52 //    for(scanf("%d",&cas),cass=1;cass<=cas;cass++)
53     while(~scanf("%d",&n))
54     {
55         lll=0;sz[0]=0;k=0;
56         for(i=1;i<=n;i++)fa[i]=i;
57         for(i=1;i<=n;i++)
58         {
59             scanf("%d%d%d",&x,&y,&z);
60             k=max(k,x);
61             k=max(k,y);
62             fx=zhao(x);
63             fy=zhao(y);
64             if(a[fy].size()>a[fx].size()swap(fx,fy);
65             if(z)
66             {
67                 if(Find(a[fy],fx))
68                 {
69                     sz[++lll]=i;
70                     for(j=1;j<=k;j++)fa[j]=j;
71                     for(j=1;j<=k;j++)a[j].clear();
72                     k=0;
73                 }
74                 else
75                 {
76                     fa[fy]=fx;
77                     a[fx].insert(a[fy].begin(),a[fy].end());
78                 }
79             }
80             else
81             {
82                 if(fx==fy)
83                 {
84                     sz[++lll]=i;
85                     for(j=1;j<=k;j++)fa[j]=j;
86                     for(j=1;j<=k;j++)a[j].clear();
87                     k=0;
88                 }
89                 else
90                 {
91                     a[fx].insert(fy);
92                     a[fy].insert(fx);
93                 }
94             }
95         }
96         for(j=1;j<=k;j++)a[j].clear();
97         printf("%d\n",lll);
98         for(i=1;i<=lll;i++)
99             printf("%d\n",sz[i]-sz[i-1]);
100     }
101     return 0;
102 }
103 /*
104 //
105
106 //
107 */

View Code