Description
小w来到百度之星的赛场上,准备开始实现一个程序自动分析系统。
这个程序接受一些形如xi=xj 或 xi≠xj 的相等/不等约束条件作为输入,判定是否可以通过给每个 w 赋适当的值,来满足这些条件。
输入包含多组数据。
然而粗心的小w不幸地把每组数据之间的分隔符删掉了。
他只知道每组数据都是不可满足的,且若把每组数据的最后一个约束条件去掉,则该组数据是可满足的。
请帮助他恢复这些分隔符。
Input
第1行:一个数字L,表示后面输入的总行数。
之后L行,每行包含三个整数, i , j , e i,j,e i,j,e,描述一个相等/不等的约束条件,若 e = 1 e=1 e=1,则该约束条件为 x i = x j x_i=x_j xi=xj ,若 e = 0 e=0 e=0,则该约束条件为 x i ≠ x j x_i≠x_j xi=xj 。
i , j , L ≤ 100000 i,j,L≤100000 i,j,L≤100000
x i , x j ≤ L x_i,x_j≤L xi,xj≤L
Output
输出共T+1行。
第一行一个整数T,表示数据组数。
接下来T行的第 i i 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
思路
用并查集来表示相等关系,对每个元素单独建立一个set来记录不等关系,由于相等关系是传递的,每次得到两个元素相等,则检查它们所在的并查集树根对应的set,看是否冲突,不冲突则合并这两棵树,同时合并set,冲突则代表找到了一组数据。每次得到两个元素不等,则检查它们是否在同一个并查集,若在则代表找到了一组数据,否则分别将冲突关系记入树根所对应的set。
AC代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
set<int>s[100005];
int f[100005];
int que[200005],a,b,c,cot;
int an[100005],ans_cot;
int findf(int now)
{
if(f[now]==now)return now;else return f[now]=findf(f[now]);
}
void clear()
{
for(int i=1;i<=cot;i++)
{
set<int>().swap(s[que[i]]);
f[que[i]]=que[i];
}
cot=0;
}
int judge_equal(int l,int r)
{
if(findf(l)==findf(r))return 0;
if(s[findf(l)].size()<s[findf(r)].size()) swap(l,r);
for(set<int>::iterator it=s[findf(r)].begin();it!=s[findf(r)].end();it++)
{
if(findf(*it)==findf(l))return 1;
}
for(set<int>::iterator it=s[findf(r)].begin();it!=s[findf(r)].end();it++)
{
s[findf(l)].insert(*it);
}
set<int>().swap(s[findf(r)]);
f[findf(r)]=f[findf(l)];
return 0;
}
int judge_unequal(int l,int r)
{
if(findf(l)==findf(r))return 1;
s[findf(l)].insert(r);
s[findf(r)].insert(l);
return 0;
}
int main()
{
int l;
for(int i=1;i<=100000;i++) f[i]=i;
scanf("%d",&l);
for(int i=1;i<=l;i++)
{
scanf("%d%d%d",&a,&b,&c);
que[++cot]=a;
que[++cot]=b;
int temp;
if(c) temp=judge_equal(a,b);else temp=judge_unequal(a,b);
if(temp)
{
an[ans_cot++]=cot/2;clear();
}
}
printf("%d\n",ans_cot);
for(int i=0;i<ans_cot;i++)printf("%d\n",an[i]);
return 0;
}