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

 

 

数据分割

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 1119    Accepted Submission(s): 268


Problem Description
小w来到百度之星的赛场上，准备开始实现一个程序自动分析系统。

这个程序接受一些形如xi=xj 或 xi≠xj 的相等/不等约束条件作为输入，判定是否可以通过给每个 w 赋适当的值，来满足这些条件。

输入包含多组数据。
然而粗心的小w不幸地把每组数据之间的分隔符删掉了。
他只知道每组数据都是不可满足的，且若把每组数据的最后一个约束条件去掉，则该组数据是可满足的。

请帮助他恢复这些分隔符。
 

Input
第1行：一个数字L，表示后面输入的总行数。

之后L行，每行包含三个整数，i,j,e，描述一个相等/不等的约束条件，若e=1，则该约束条件为xi=xj ，若e=0，则该约束条件为 xi≠xj 。

i,j,L≤100000

xi,xj≤L
 

Output
输出共T+1行。

第一行一个整数T，表示数据组数。

接下来T行的第i行，一个整数，表示第i组数据中的约束条件个数。
 

Sample Input
6 2 2 1 2 2 1 1 1 1 3 1 1 1 3 1 1 3 0
 

Sample Output
1 6
 

Source
2017"百度之星"程序设计大赛 - 初赛（A）
 

Recommend
liuyiding   |   We have carefully selected several similar problems for you:  6113 6112 6111 6110 6109 
 

Statistic | Submit | Discuss | Note

 

题目链接：

　　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[N],sz[N];
 28 set<int>a[N];
 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

 

posted @ 2017-08-13 14:00  Cool639zhu  阅读(524)  评论(0)  编辑  收藏  举报