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

数据分割

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 中。

/****************************************************

    Author : Coolxxx
    Copyright 2017 by Coolxxx. All rights reserved.
    BLOG : http://blog.csdn.net/u010568270

****************************************************/
#include<bits/stdc++.h>
#pragma comment(linker,"/STACK:1024000000,1024000000")
#define abs(a) ((a)>0?(a):(-(a)))
#define lowbit(a) (a&(-a))
#define sqr(a) ((a)*(a))
#define mem(a,b) memset(a,b,sizeof(a))
const double EPS=0.00001;
const int J=10;
const int MOD=100000007;
const int MAX=0x7f7f7f7f;
const double PI=3.14159265358979323;
const int N=100004;
const int M=1004;
using namespace std;
typedef long long LL;
double anss;
LL aans;
int cas,cass;
int n,m,lll,ans;
int fa[N],sz[N];
set<int>a[N];
int zhao(int aa)
{
    if(fa[aa]==aa || fa[aa]==0)return aa;
    return fa[aa]=zhao(fa[aa]);
}
bool Find(set<int>q,int x)
{
    set<int>::iterator it;
    for(it=q.begin();it!=q.end();it++)
    {
        if(zhao(*it)==x)return 1;
    }
    return 0;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("1.txt","r",stdin);
//    freopen("2.txt","w",stdout);
    #endif
    int i,j,k;
    int x,y,z,fx,fy;
//    for(scanf("%d",&cass);cass;cass--)
//    for(scanf("%d",&cas),cass=1;cass<=cas;cass++)
    while(~scanf("%d",&n))
    {
        lll=0;sz[0]=0;k=0;
        for(i=1;i<=n;i++)fa[i]=i;
        for(i=1;i<=n;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            k=max(k,x);
            k=max(k,y);
            fx=zhao(x);
            fy=zhao(y);
            if(a[fy].size()>a[fx].size()swap(fx,fy);
            if(z)
            {
                if(Find(a[fy],fx))
                {
                    sz[++lll]=i;
                    for(j=1;j<=k;j++)fa[j]=j;
                    for(j=1;j<=k;j++)a[j].clear();
                    k=0;
                }
                else
                {
                    fa[fy]=fx;
                    a[fx].insert(a[fy].begin(),a[fy].end());
                }
            }
            else
            {
                if(fx==fy)
                {
                    sz[++lll]=i;
                    for(j=1;j<=k;j++)fa[j]=j;
                    for(j=1;j<=k;j++)a[j].clear();
                    k=0;
                }
                else 
                {
                    a[fx].insert(fy);
                    a[fy].insert(fx);
                }
            }
        }
        for(j=1;j<=k;j++)a[j].clear();
        printf("%d
",lll);
        for(i=1;i<=lll;i++)
            printf("%d
",sz[i]-sz[i-1]);
    }
    return 0;
}
/*
//

//
*/