HDU6109 并查集&set

数据分割

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

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

 

中文题面，题目大意很清楚2333

因为这个题被挂在启发式合并与左偏树专题下，感到非常懵逼，感觉并查集可以搞定emmm没想到真的emmm；

思路：

考虑相等关系与不等关系的存储，相等关系显然可以用并查集来存储，至于不等关系，想到开set数组存储各节点不等关系。

需要注意，一棵相等集合，他们不等的集合也应合并考虑，一棵相等树中不相同的集合应是一个！合并到根节点！

 

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
using namespace std;

const int maxn=1e5+10;

int n;
int p[maxn];
set <int> s[maxn];

int t;
int ans[maxn];

//初始化
void init()
{
	for(int i=1;i<=n;++i){
		p[i]=i;
		s[i].clear();
	}
}

int find(int x)
{
	return x==p[x]?x:p[x]=find(p[x]);
}

int main()
{
	while(~scanf("%d",&n)){
		int pre=0;
		memset(ans,0,sizeof(ans));
		init();
		for(int l=1;l<=n;++l){
			int i,j,e;
			scanf("%d%d%d",&i,&j,&e);
			int pi=find(i),pj=find(j);
			if(e){
				if(pi==pj) continue;
				else if(!s[pi].count(pj)){
					//注，不只是简单的p[y]=x;
					//一棵相等树中不相同的集合应是一个！合并到根节点！
					for(set<int>::iterator it=s[pj].begin();it!=s[pj].end();it++){
						s[pi].insert(*it);
					}
					p[pj]=pi;
				}
				else{
					ans[++t]=l-pre;
					init();
					pre=l;
				}
			}
			else{
				if(pi==pj){
					ans[++t]=l-pre;
					init();
					pre=l;
				}
				else{
					s[i].insert(j);
					s[j].insert(i);
				}
			}
		}
		printf("%d\n",t);
		for(int i=1;i<=t;++i){
			printf("%d\n",ans[i]);
		}
	}

	return 0;
}